nodes/taabata-SANA_LOWVRAM
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
taabata
2024-12-28 04:16:29 +03:00
committed by GitHub
parent 9adf754a6b
commit cc1d970d9b
1 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+17 -5
View File
@@ -9,6 +9,12 @@ from PIL import Image
class SANADiffuse:
models = ["Efficient-Large-Model/Sana_600M_512px_diffusers",
"Efficient-Large-Model/Sana_600M_1024px_diffusers",
"Efficient-Large-Model/Sana_1600M_512px_diffusers",
"Efficient-Large-Model/Sana_1600M_1024px_diffusers"
]
@classmethod
def INPUT_TYPES(s):
return {"required":
@@ -47,7 +53,7 @@ class SANADiffuse:
}),
"img2img":(["disable","enable"],),
"embeds":("class",),
"model_path": ([f'diffusers/{i}' for i in os.listdir(folder_paths.get_folder_paths("diffusers")[0]) if os.path.isdir(folder_paths.get_folder_paths("diffusers")[0]+f"/{i}")],),
"model_path": ([f'diffusers/{i}' for i in os.listdir(folder_paths.get_folder_paths("diffusers")[0]) if os.path.isdir(folder_paths.get_folder_paths("diffusers")[0]+f"/{i}")]+SANADiffuse.models,),
"device":(["cuda","cpu"],),
},
"optional":
@@ -87,7 +93,7 @@ class SANADiffuse:
"img2img":img2img,
"image":image,
"strength":strength,
"model":os.path.dirname(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))+f"/models/{model_path}"
"model":os.path.dirname(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))+f"/models/{model_path}" if model_path not in SANADiffuse.models else model_path
}
req = requests.post("http://127.0.0.1:5000/diffuse",json=data)
while flag:
@@ -108,13 +114,19 @@ class SANADiffuse:
class SANATextEncode:
models = ["Efficient-Large-Model/Sana_600M_512px_diffusers",
"Efficient-Large-Model/Sana_600M_1024px_diffusers",
"Efficient-Large-Model/Sana_1600M_512px_diffusers",
"Efficient-Large-Model/Sana_1600M_1024px_diffusers"
]
@classmethod
def INPUT_TYPES(s):
return {"required":
{
"prompt": ("STRING", {"default": '', "multiline": True}),
"negative_prompt": ("STRING", {"default": '', "multiline": True}),
"model_path": ([f'diffusers/{i}' for i in os.listdir(folder_paths.get_folder_paths("diffusers")[0]) if os.path.isdir(folder_paths.get_folder_paths("diffusers")[0]+f"/{i}")],),
"model_path": ([f'diffusers/{i}' for i in os.listdir(folder_paths.get_folder_paths("diffusers")[0]) if os.path.isdir(folder_paths.get_folder_paths("diffusers")[0]+f"/{i}")]+SANATextEncode.models,),
}
}
@@ -129,7 +141,7 @@ class SANATextEncode:
data = {
"prompt":prompt,
"negative_prompt":negative_prompt,
"model":os.path.dirname(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))+f"/models/{model_path}"
"model":os.path.dirname(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))+f"/models/{model_path}" if model_path not in SANATextEncode.models else model_path
}
requests.post("http://127.0.0.1:5000/encode",json=data)
flag = True
@@ -151,4 +163,4 @@ class SANATextEncode:
NODE_CLASS_MAPPINGS = {
"SANADiffuse":SANADiffuse,
"SANATextEncode":SANATextEncode
}
}