Spaces:
Running
Running
File size: 2,299 Bytes
91fb4ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
import os
DEFAULT_HEIGHT_BUCKETS = [256, 320, 384, 480, 512, 576, 720, 768, 960, 1024, 1280, 1536]
DEFAULT_WIDTH_BUCKETS = [256, 320, 384, 480, 512, 576, 720, 768, 960, 1024, 1280, 1536]
DEFAULT_FRAME_BUCKETS = [49]
DEFAULT_IMAGE_RESOLUTION_BUCKETS = []
for height in DEFAULT_HEIGHT_BUCKETS:
for width in DEFAULT_WIDTH_BUCKETS:
DEFAULT_IMAGE_RESOLUTION_BUCKETS.append((height, width))
DEFAULT_VIDEO_RESOLUTION_BUCKETS = []
for frames in DEFAULT_FRAME_BUCKETS:
for height in DEFAULT_HEIGHT_BUCKETS:
for width in DEFAULT_WIDTH_BUCKETS:
DEFAULT_VIDEO_RESOLUTION_BUCKETS.append((frames, height, width))
FINETRAINERS_LOG_LEVEL = os.environ.get("FINETRAINERS_LOG_LEVEL", "INFO")
PRECOMPUTED_DIR_NAME = "precomputed"
PRECOMPUTED_CONDITIONS_DIR_NAME = "conditions"
PRECOMPUTED_LATENTS_DIR_NAME = "latents"
MODEL_DESCRIPTION = r"""
\# {model_id} {training_type} finetune
<Gallery />
\#\# Model Description
This model is a {training_type} of the `{model_id}` model.
This model was trained using the `fine-video-trainers` library - a repository containing memory-optimized scripts for training video models with [Diffusers](https://github.com/huggingface/diffusers).
\#\# Download model
[Download LoRA]({repo_id}/tree/main) in the Files & Versions tab.
\#\# Usage
Requires [🧨 Diffusers](https://github.com/huggingface/diffusers) installed.
```python
{model_example}
```
For more details, including weighting, merging and fusing LoRAs, check the [documentation](https://huggingface.co/docs/diffusers/main/en/using-diffusers/loading_adapters) on loading LoRAs in diffusers.
\#\# License
Please adhere to the license of the base model.
""".strip()
_COMMON_BEGINNING_PHRASES = (
"This video",
"The video",
"This clip",
"The clip",
"The animation",
"This image",
"The image",
"This picture",
"The picture",
)
_COMMON_CONTINUATION_WORDS = ("shows", "depicts", "features", "captures", "highlights", "introduces", "presents")
COMMON_LLM_START_PHRASES = (
"In the video,",
"In this video,",
"In this video clip,",
"In the clip,",
"Caption:",
*(
f"{beginning} {continuation}"
for beginning in _COMMON_BEGINNING_PHRASES
for continuation in _COMMON_CONTINUATION_WORDS
),
)
|