Spaces:
Runtime error
Runtime error
Commit
•
bfea134
1
Parent(s):
d8b8452
fix pipe
Browse files- sd_model.py +1 -1
sd_model.py
CHANGED
|
@@ -42,7 +42,7 @@ def get_cn_detector(image):
|
|
| 42 |
print(detectors)
|
| 43 |
return detectors
|
| 44 |
|
| 45 |
-
@spaces.GPU
|
| 46 |
def generate(pipe, detectors, prompt, negative_prompt, reference_flg=False, reference_img=None):
|
| 47 |
pipe.to("cuda")
|
| 48 |
default_pos = ""
|
|
|
|
| 42 |
print(detectors)
|
| 43 |
return detectors
|
| 44 |
|
| 45 |
+
@spaces.GPU(duration=120)
|
| 46 |
def generate(pipe, detectors, prompt, negative_prompt, reference_flg=False, reference_img=None):
|
| 47 |
pipe.to("cuda")
|
| 48 |
default_pos = ""
|