diff --git a/nodes.py b/nodes.py index 976ca0a..9dd6e05 100644 --- a/nodes.py +++ b/nodes.py @@ -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" -} \ No newline at end of file +}