Update model.py

model.py

@@ -187,7 +187,7 @@ class Model:
         
         with torch.no_grad():
             # extract condition features
-            features_adapter = self.model_ad_pose(im.to(device))
+            features_adapter = self.model_ad_pose(pose.to(device))
     
         shape = [4, 64, 64]