Spaces:
Runtime error
Runtime error
Commit
•
c8d48fa
1
Parent(s):
4e7fe8d
fix pipe
Browse files- sd_model.py +1 -0
sd_model.py
CHANGED
|
@@ -52,6 +52,7 @@ def get_cn_detector(image):
|
|
| 52 |
|
| 53 |
@spaces.GPU
|
| 54 |
def generate(pipe, detectors, prompt, negative_prompt):
|
|
|
|
| 55 |
default_pos = ""
|
| 56 |
default_neg = ""
|
| 57 |
prompt = default_pos + prompt
|
|
|
|
| 52 |
|
| 53 |
@spaces.GPU
|
| 54 |
def generate(pipe, detectors, prompt, negative_prompt):
|
| 55 |
+
pipe.to("cuda")
|
| 56 |
default_pos = ""
|
| 57 |
default_neg = ""
|
| 58 |
prompt = default_pos + prompt
|