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
|