Merge pull request #23 from sparkleMing/fix/da_model_referenced_before_assignment

fix Local variable 'da_model' might be referenced before assignment
2025-03-06 14:01:52 +02:00
parent 003d7b44ba b3bd8499e7
commit 9d7cb8c1e5
1 changed files with 7 additions and 6 deletions
@@ -105,11 +105,12 @@ fp16 reduces quality by a LOT, not recommended.
                self.model.load_state_dict(state_dict)

            self.model.eval()
-            da_model = {
-                "model": self.model,
-                "dtype": dtype,
-                "is_metric": self.model.is_metric
-            }
+ 
+        da_model = {
+            "model": self.model,
+            "dtype": dtype,
+            "is_metric": self.model.is_metric
+        }
           
        return (da_model,)
    
@@ -185,4 +186,4 @@ NODE_CLASS_MAPPINGS = {
 NODE_DISPLAY_NAME_MAPPINGS = {
    "DepthAnything_V2": "Depth Anything V2",
    "DownloadAndLoadDepthAnythingV2Model": "DownloadAndLoadDepthAnythingV2Model"
-}
+}