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
|