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):
|