Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -46,14 +46,14 @@ repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="defaul
|
|
| 46 |
repo_default.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="pixel_base")
|
| 47 |
repo_default.load_lora_weights("nerijs/pixel-art-xl", adapter_name="pixel_base_2")
|
| 48 |
|
| 49 |
-
|
| 50 |
|
| 51 |
repo_customs = {
|
| 52 |
"Default": repo_default,
|
| 53 |
"Realistic": DiffusionPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
|
| 54 |
"Anime": DiffusionPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
|
| 55 |
"Pixel": repo_default,
|
| 56 |
-
"Pro":
|
| 57 |
}
|
| 58 |
|
| 59 |
# Functions
|
|
|
|
| 46 |
repo_default.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="pixel_base")
|
| 47 |
repo_default.load_lora_weights("nerijs/pixel-art-xl", adapter_name="pixel_base_2")
|
| 48 |
|
| 49 |
+
repo_pro = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, use_safetensors=True, add_watermarker=False)
|
| 50 |
|
| 51 |
repo_customs = {
|
| 52 |
"Default": repo_default,
|
| 53 |
"Realistic": DiffusionPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
|
| 54 |
"Anime": DiffusionPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
|
| 55 |
"Pixel": repo_default,
|
| 56 |
+
"Pro": repo_pro,
|
| 57 |
}
|
| 58 |
|
| 59 |
# Functions
|