Jordan Legg commited on
Commit
b029f93
·
1 Parent(s): ee2ff4c

added trigger word

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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()