Spaces:
Running
on
Zero
Running
on
Zero
Jordan Legg
commited on
Commit
·
3dd7686
1
Parent(s):
b029f93
commented out LoRA, needs it's own LoRA ideally
Browse files
app.py
CHANGED
|
@@ -11,8 +11,8 @@ device: str = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 11 |
|
| 12 |
|
| 13 |
pipe = DiffusionPipeline.from_pretrained("shuttleai/shuttle-3-diffusion", torch_dtype=dtype).to(device)
|
| 14 |
-
trigger_word = "Ultra realistic"
|
| 15 |
-
pipe.load_lora_weights("prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0")
|
| 16 |
# Enable VAE tiling
|
| 17 |
pipe.vae.enable_tiling()
|
| 18 |
|
|
|
|
| 11 |
|
| 12 |
|
| 13 |
pipe = DiffusionPipeline.from_pretrained("shuttleai/shuttle-3-diffusion", torch_dtype=dtype).to(device)
|
| 14 |
+
# trigger_word = "Ultra realistic"
|
| 15 |
+
# pipe.load_lora_weights("prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0")
|
| 16 |
# Enable VAE tiling
|
| 17 |
pipe.vae.enable_tiling()
|
| 18 |
|