Spaces:
Sleeping
Sleeping
Jordan Legg
commited on
Commit
·
b029f93
1
Parent(s):
ee2ff4c
added trigger word
Browse files
app.py
CHANGED
@@ -11,6 +11,7 @@ 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 |
pipe.load_lora_weights("prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0")
|
15 |
# Enable VAE tiling
|
16 |
pipe.vae.enable_tiling()
|
|
|
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()
|