Fix merge

comfy/model_management.py

@@ -2,6 +2,7 @@
 import logging
 from enum import Enum
 from comfy.cli_args import args
+import comfy
 import torch
 import sys
 import platform