Update model.py

model.py

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