prithivMLmods commited on
Commit
47a038c
Β·
verified Β·
1 Parent(s): e259a1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,8 +26,8 @@ if not torch.cuda.is_available():
26
  base_model = "black-forest-labs/FLUX.1-dev"
27
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
28
 
29
- lora_repo = "strangerzonehf/Flux-C7-Sketch-LoRA"
30
- trigger_word = "C7 Sketch" # Leave trigger_word blank if not used.
31
 
32
  pipe.load_lora_weights(lora_repo)
33
  pipe.to("cuda")
 
26
  base_model = "black-forest-labs/FLUX.1-dev"
27
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
28
 
29
+ lora_repo = "strangerzonehf/Flux-Sketch-Sized-LoRA"
30
+ trigger_word = "Sketch Sized" # Leave trigger_word blank if not used.
31
 
32
  pipe.load_lora_weights(lora_repo)
33
  pipe.to("cuda")