| 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] | |
| ] | |