Maria
commited on
Commit
·
bcad7c8
1
Parent(s):
5ac6617
hw6
Browse files
infer.py
CHANGED
|
@@ -165,6 +165,8 @@ def prepare_controlnet_image(controlnet_image, mode):
|
|
| 165 |
else:
|
| 166 |
image = controlnet_image
|
| 167 |
|
|
|
|
|
|
|
| 168 |
# @spaces.GPU #[uncomment to use ZeroGPU]
|
| 169 |
def infer(
|
| 170 |
model_id,
|
|
|
|
| 165 |
else:
|
| 166 |
image = controlnet_image
|
| 167 |
|
| 168 |
+
return image
|
| 169 |
+
|
| 170 |
# @spaces.GPU #[uncomment to use ZeroGPU]
|
| 171 |
def infer(
|
| 172 |
model_id,
|