xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-06-01 23:00:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add intristic loras

Browse Source
This commit is contained in:
kijai 2024-02-26 11:19:40 +02:00
parent 6829b0ccf8
commit d228e713d6
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
intrinsic_loras.txt Normal file
View File

@ -0,0 +1,4 @@
source for the loras:
https://github.com/duxiaodan/intrinsic-lora
Renamed and conveted to .safetensors

7
nodes.py
View File

@ -3663,15 +3663,17 @@ class ImageNormalize_Neg1_To_1:
import comfy.sample import comfy.sample
from nodes import CLIPTextEncode, VAEEncode, VAEDecode, LoraLoader from nodes import CLIPTextEncode, LoraLoader
folder_paths.add_model_folder_path("intristic_loras", os.path.join(script_dir, "intristic_loras"))
class Intrinsic_lora_sampling: class Intrinsic_lora_sampling:
def __init__(self): def __init__(self):
self.loaded_lora = None self.loaded_lora = None
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": { "model": ("MODEL",), return {"required": { "model": ("MODEL",),
"lora_name": (folder_paths.get_filename_list("loras"), ), "lora_name": (folder_paths.get_filename_list("intristic_loras"), ),
"task": ( "task": (
[ [
'depth map', 'depth map',
@ -3735,7 +3737,6 @@ class Intrinsic_lora_sampling:
for start_idx in range(0, samples["samples"].shape[0], per_batch): for start_idx in range(0, samples["samples"].shape[0], per_batch):
decoded.append(vae.decode(samples["samples"][start_idx:start_idx+per_batch])) decoded.append(vae.decode(samples["samples"][start_idx:start_idx+per_batch]))
image_out = torch.cat(decoded, dim=0) image_out = torch.cat(decoded, dim=0)
#image_out, = VAEDecode.decode(self, vae, samples)
pbar.update(1) pbar.update(1)