Jamal811 commited on
Commit
7b9afe1
·
verified ·
1 Parent(s): 12f179f
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ refiner = DiffusionPipeline.from_pretrained(
17
  use_safetensors=True,
18
  variant="fp16"
19
  )
20
- refiner.to("cuda") # Use GPU if available, else change to "cpu"
21
 
22
  # Function to generate the image
23
  def generate_image(prompt, n_steps=20, high_noise_frac=0.8):
 
17
  use_safetensors=True,
18
  variant="fp16"
19
  )
20
+ refiner.to("cpu") # Use GPU if available, else change to "cpu"
21
 
22
  # Function to generate the image
23
  def generate_image(prompt, n_steps=20, high_noise_frac=0.8):