Spaces:
Runtime error
Runtime error
Commit
·
c242bd5
1
Parent(s):
631d3bb
Update model.py
Browse files
model.py
CHANGED
|
@@ -187,7 +187,7 @@ class Model:
|
|
| 187 |
|
| 188 |
with torch.no_grad():
|
| 189 |
# extract condition features
|
| 190 |
-
features_adapter = self.model_ad_pose(
|
| 191 |
|
| 192 |
shape = [4, 64, 64]
|
| 193 |
|
|
|
|
| 187 |
|
| 188 |
with torch.no_grad():
|
| 189 |
# extract condition features
|
| 190 |
+
features_adapter = self.model_ad_pose(pose.to(device))
|
| 191 |
|
| 192 |
shape = [4, 64, 64]
|
| 193 |
|