nodes/kijai-ComfyUI-PyramidFlowWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai
2024-11-07 21:50:56 +02:00
parent 3aeea32237
commit 1d9fe8c695
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+4 -1
View File
@@ -224,6 +224,7 @@ class PyramidFlowModelLoader:
return (pyramid_model,) return (pyramid_model,)
#region Sampler
class PyramidFlowSampler: class PyramidFlowSampler:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
@@ -358,7 +359,7 @@ class PyramidFlowTextEncode:
if force_offload: if force_offload:
clip.cond_stage_model.to(offload_device) clip.cond_stage_model.to(offload_device)
clip.cond_stage_model.reset_clip_options()
embeds = { embeds = {
"prompt_embeds": prompt_embeds.to(device), "prompt_embeds": prompt_embeds.to(device),
"attention_mask": prompt_attention_mask["attention_mask"].to(device), "attention_mask": prompt_attention_mask["attention_mask"].to(device),
@@ -369,6 +370,8 @@ class PyramidFlowTextEncode:
} }
return (embeds, ) return (embeds, )
#region VAE
class PyramidFlowVAEEncode: class PyramidFlowVAEEncode:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):