Hugging Face's logo

Spaces:
Dagfinn1962
/
FLUX-REALISM
Running on Zero

App Files Community
prithivMLmods commited on
Commit
49393d8
·
verified ·
1 Parent(s): fcc63e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
40
  lora_repo = "prithivMLmods/Fashion-Hut-Modeling-LoRA"
41
  trigger_word = "Modeling of" # Leave trigger_word blank if not used.
42
  pipe.load_lora_weights(lora_repo)
43
-
44
  #lora_repo = "prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0"
45
  #trigger_word = "Ultra realistic" # Leave trigger_word blank if not used.
46
  #pipe.load_lora_weights(lora_repo)
 
40
  lora_repo = "prithivMLmods/Fashion-Hut-Modeling-LoRA"
41
  trigger_word = "Modeling of" # Leave trigger_word blank if not used.
42
  pipe.load_lora_weights(lora_repo)
43
+ pipe.to("cuda")
44
  #lora_repo = "prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0"
45
  #trigger_word = "Ultra realistic" # Leave trigger_word blank if not used.
46
  #pipe.load_lora_weights(lora_repo)