Update app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ repo_pixel.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="ba
|
|
| 53 |
repo_pixel.load_lora_weights("nerijs/pixel-art-xl", adapter_name="base2")
|
| 54 |
repo_pixel.set_adapters(["base", "base2"], adapter_weights=[1, 1])
|
| 55 |
|
| 56 |
-
repo_neo = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, use_safetensors=True, add_watermarker=False, revision="refs/pr/1")
|
| 57 |
repo_neo.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
| 58 |
repo_neo.set_adapters(["base"], adapter_weights=[0.7])
|
| 59 |
|
|
|
|
| 53 |
repo_pixel.load_lora_weights("nerijs/pixel-art-xl", adapter_name="base2")
|
| 54 |
repo_pixel.set_adapters(["base", "base2"], adapter_weights=[1, 1])
|
| 55 |
|
| 56 |
+
repo_neo = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, use_safetensors=True, add_watermarker=False, revision="refs/pr/1").to(DEVICE)
|
| 57 |
repo_neo.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
| 58 |
repo_neo.set_adapters(["base"], adapter_weights=[0.7])
|
| 59 |
|