Pijush2023 commited on
Commit
48b772e
·
verified ·
1 Parent(s): 183be31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -706,7 +706,7 @@ lora_repo = "XLabs-AI/flux-RealismLora"
706
  trigger_word = "" # Leave trigger_word blank if not used.
707
  pipe.load_lora_weights(lora_repo)
708
 
709
- pipe.to("cpu")
710
 
711
  MAX_SEED = 2**32-1
712
 
 
706
  trigger_word = "" # Leave trigger_word blank if not used.
707
  pipe.load_lora_weights(lora_repo)
708
 
709
+ pipe.to("cuda")
710
 
711
  MAX_SEED = 2**32-1
712