import comfy.model_management

class AnyType(str):
  """A special class that is always equal in not equal comparisons. Credit to pythongosssss"""

  def __ne__(self, __value: object) -> bool:
    return False


any = AnyType("*")

class UnloadAllModels:
    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "any":(any,{"forceInput": True})
            },
            "optional": {},
        }

    RETURN_TYPES = ()
    FUNCTION = "unload_models"
    CATEGORY = "不忘科技-自定义节点🚩"
    OUTPUT_NODE = True

    def unload_models(self,any=None):
        # 卸载所有已加载的模型
        comfy.model_management.soft_empty_cache()
        comfy.model_management.unload_all_models()
        comfy.model_management.soft_empty_cache()
        return ()