Spaces:
Runtime error
Runtime error
Commit
·
aff3d3b
1
Parent(s):
03bc7e7
Update model.py
Browse files
model.py
CHANGED
|
@@ -133,7 +133,7 @@ class Model:
|
|
| 133 |
|
| 134 |
with torch.no_grad():
|
| 135 |
# extract condition features
|
| 136 |
-
features_adapter = self.
|
| 137 |
|
| 138 |
shape = [4, 64, 64]
|
| 139 |
|
|
|
|
| 133 |
|
| 134 |
with torch.no_grad():
|
| 135 |
# extract condition features
|
| 136 |
+
features_adapter = self.model_ad_sketch(im.to(device))
|
| 137 |
|
| 138 |
shape = [4, 64, 64]
|
| 139 |
|