File size: 556 Bytes
d65b1bc acc7164 d65b1bc cb4ba01 acc7164 d685a1d |
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 |
from typing import List, Optional
from pydantic import BaseModel, HttpUrl
from pydantic import validator
class Scene(BaseModel):
narration: str
image_prompts: List[str]
voice: str
class Story(BaseModel):
scenes: List[Scene]
reference_image_url: Optional[str] = None
ip_adapter_weight: Optional[float] = 0.4 # Add this line
class GeneratorRequest(BaseModel):
prompt: str
batch_size: int = 4
grok: bool = False
model: str = "command"
class GeneratorBulkRequest(BaseModel):
stories: List[GeneratorRequest]
|