Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

wip: basic wrapper for generating sd3 images

Browse Source
This commit is contained in:
blessedcoolant
2024-06-16 04:18:20 +05:30
parent 554809c647
commit f65d50a4dd
10 changed files with 452 additions and 173 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
invokeai/app/invocations/model.py
View File

@ -60,6 +60,15 @@ class CLIPField(BaseModel):
loras: List[LoRAField] = Field(description="LoRAs to apply on model loading")
class SD3CLIPField(BaseModel):
tokenizer_1: ModelIdentifierField = Field(description="Info to load tokenizer 1 submodel")
text_encoder_1: ModelIdentifierField = Field(description="Info to load text_encoder 1 submodel")
tokenizer_2: ModelIdentifierField = Field(description="Info to load tokenizer 2 submodel")
text_encoder_2: ModelIdentifierField = Field(description="Info to load text_encoder 2 submodel")
tokenizer_3: ModelIdentifierField = Field(description="Info to load tokenizer 3 submodel")
text_encoder_3: ModelIdentifierField = Field(description="Info to load text_encoder 3 submodel")
class VAEField(BaseModel):
vae: ModelIdentifierField = Field(description="Info to load vae submodel")
seamless_axes: List[str] = Field(default_factory=list, description='Axes("x" and "y") to which apply seamless')