CyranoB commited on
Commit
3d20680
·
1 Parent(s): 3678910

Trying to add CPU support

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -113,7 +113,7 @@ def check_nsfw_images(
113
  return images, has_nsfw_concepts
114
 
115
  # Function
116
- @spaces.GPU(enable_queue=True)
117
  def generate_image(prompt, ckpt, aspect_ratio, mode):
118
  width, height = calculate_resolution(aspect_ratio, mode) # Calculate resolution based on the aspect ratio
119
  checkpoint = checkpoints[ckpt][0]
 
113
  return images, has_nsfw_concepts
114
 
115
  # Function
116
+ #@spaces.GPU(enable_queue=True)
117
  def generate_image(prompt, ckpt, aspect_ratio, mode):
118
  width, height = calculate_resolution(aspect_ratio, mode) # Calculate resolution based on the aspect ratio
119
  checkpoint = checkpoints[ckpt][0]