Spaces:

RamAnanth1
/

T2I-Adapter

Runtime error

RamAnanth1 commited on Feb 20, 2023

Commit

9022b97

•

1 Parent(s): 8b3abf5

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -308,7 +308,7 @@ class Model:
         # optional
         return_heatmap = False
-        dataset = pose_model.cfg.data['test']['type']
         # e.g. use ('backbone', ) to return backbone feature
         output_layer_names = None

         # optional
         return_heatmap = False
+        dataset = self.pose_model.cfg.data['test']['type']
         # e.g. use ('backbone', ) to return backbone feature
         output_layer_names = None