Update app.py
Browse files
app.py
CHANGED
|
@@ -62,15 +62,10 @@ repo_customs = {
|
|
| 62 |
"Realistic": None, #StableDiffusionXLPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
|
| 63 |
"Anime": None, #StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
|
| 64 |
"Pixel": None, #repo_pixel,
|
| 65 |
-
"Large": StableDiffusion3Pipeline.from_pretrained(repo_large_path, transformer=repo_large_transformer, torch_dtype=torch.float16, use_safetensors=True,
|
| 66 |
-
text_encoder_3=None,
|
| 67 |
-
tokenizer_3=None),
|
| 68 |
}
|
| 69 |
|
| 70 |
-
repo_customs["Large"].scheduler = FlashFlowMatchEulerDiscreteScheduler.from_pretrained(
|
| 71 |
-
repo_large_path,
|
| 72 |
-
subfolder="scheduler",
|
| 73 |
-
)
|
| 74 |
|
| 75 |
# Functions
|
| 76 |
def save_image(img, seed):
|
|
|
|
| 62 |
"Realistic": None, #StableDiffusionXLPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
|
| 63 |
"Anime": None, #StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
|
| 64 |
"Pixel": None, #repo_pixel,
|
| 65 |
+
"Large": StableDiffusion3Pipeline.from_pretrained(repo_large_path, transformer=repo_large_transformer, torch_dtype=torch.float16, use_safetensors=True),
|
|
|
|
|
|
|
| 66 |
}
|
| 67 |
|
| 68 |
+
repo_customs["Large"].scheduler = FlashFlowMatchEulerDiscreteScheduler.from_pretrained(repo_large_path, subfolder="scheduler")
|
|
|
|
|
|
|
|
|
|
| 69 |
|
| 70 |
# Functions
|
| 71 |
def save_image(img, seed):
|