uttamg07 commited on
Commit
251b17c
·
1 Parent(s): b31c073

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ pipe = DiffusionPipeline.from_pretrained(
28
  )
29
 
30
  pipe.load_lora_weights("./lora-trained-thumbs-up-pw-0.5-steps-800-uttam-pw-0.5-steps-1600")
31
- # _ = pipe.to("cuda")
32
 
33
  prompt = st.text_area("Enter the prompt!")
34
  image = pipe(prompt=prompt, num_inference_steps=25, num_images_per_prompt=1)[:]
 
28
  )
29
 
30
  pipe.load_lora_weights("./lora-trained-thumbs-up-pw-0.5-steps-800-uttam-pw-0.5-steps-1600")
31
+ _ = pipe.to("cuda")
32
 
33
  prompt = st.text_area("Enter the prompt!")
34
  image = pipe(prompt=prompt, num_inference_steps=25, num_images_per_prompt=1)[:]