32 lines
817 B
Python
32 lines
817 B
Python
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 () |