Spaces:
Running
Running
Trying to add CPU support
Browse files
app.py
CHANGED
|
@@ -113,7 +113,7 @@ def check_nsfw_images(
|
|
| 113 |
return images, has_nsfw_concepts
|
| 114 |
|
| 115 |
# Function
|
| 116 |
-
|
| 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]
|