nodes/kijai-ComfyUI-DynamiCrafterWrapper
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-03-16 19:30:01 +02:00
parent 0df3ba8a57
commit 02b640758e
1 changed files with 2 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+2 -16
View File
@@ -78,13 +78,6 @@ class DynamiCrafterI2V:
"prompt": ("STRING", {"multiline": True, "default": "",}),
"seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}),
"fs": ("INT", {"default": 10, "min": 2, "max": 100, "step": 1}),
"dtype": (
[
'fp32',
'fp16',
], {
"default": 'fp16'
}),
"keep_model_loaded": ("BOOLEAN", {"default": True}),
},
@@ -98,7 +91,7 @@ class DynamiCrafterI2V:
FUNCTION = "process"
CATEGORY = "DynamiCrafterWrapper"
def process(self, model, image, dtype, prompt, cfg, steps, eta, seed, fs, keep_model_loaded, frames, image2=None):
def process(self, model, image, prompt, cfg, steps, eta, seed, fs, keep_model_loaded, frames, image2=None):
device = mm.get_torch_device()
mm.unload_all_models()
mm.soft_empty_cache()
@@ -222,13 +215,6 @@ class DynamiCrafterBatchInterpolation:
"prompt": ("STRING", {"multiline": True, "default": "",}),
"seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}),
"fs": ("INT", {"default": 10, "min": 2, "max": 100, "step": 1}),
"dtype": (
[
'fp32',
'fp16',
], {
"default": 'fp16'
}),
"keep_model_loaded": ("BOOLEAN", {"default": True}),
},
@@ -239,7 +225,7 @@ class DynamiCrafterBatchInterpolation:
FUNCTION = "process"
CATEGORY = "DynamiCrafterWrapper"
def process(self, model, images, dtype, prompt, cfg, steps, eta, seed, fs, keep_model_loaded):
def process(self, model, images, prompt, cfg, steps, eta, seed, fs, keep_model_loaded):
device = mm.get_torch_device()
mm.unload_all_models()
mm.soft_empty_cache()