config.py

import os
import numpy as np

TITLE = "Mochi 1 - Video Generator 🌟"

DESCRIPTION = """  
✨ Generate a video based on a text prompt using the MochiPipeline. 🎥                                            
Enter a text prompt and customize the number of frames and the frame rate (fps) for the generated video. 🛠️     
"""

BUY_ME_A_COFFE = """  
<a href="https://buymeacoffee.com/thesab" target="_blank">  
    <button style="background-color: #FFDD00; border: none; color: black; padding: 10px 20px; text-align: center; text-decoration: none; display: inline-block; font-size: 16px; margin: 4px 2px; cursor: pointer; border-radius: 10px;">  
        ☕ Buy Me a Coffee  
    </button>  
</a>   
"""


MODEL_PRE_TRAINED_ID = "genmo/mochi-1-preview"

MAX_SEED = np.iinfo(np.int32).max
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1024"))
DEFAULT_SEED = 42
DEFAULT_RANDOMIZE_SEED = False
DEFAULT_WIDTH = 1024
DEFAULT_HEIGHT = 1024
DEFAULT_GUIDANCE_SCALE = 4.5
DEFAULT_NUM_INFERENCE_STEPS = 40
DEFAULT_LORA_SCALE = 4.5

EXAMPLES = [
     ["A beautiful sunrise over the mountains 🌄", 100, 24],
    ["A futuristic city with flying cars 🚗✨", 120, 30],
    ["A serene beach with waves crashing 🌊", 80, 25],
    ["A bustling market in a medieval town 🏰", 90, 20],
    ["A spaceship traveling through space 🚀", 110, 30]
]