mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-10 11:35:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

ltxv: add noise to guidance image to ensure generated motion. (#5937)

Browse Source
This commit is contained in:
Michael Kupchick
2024-12-06 12:46:08 +02:00
committed by GitHub
parent 1e21f4c14e
commit 005d2d3a13
2 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
comfy_extras/nodes_lt.py
View File

@@ -1,3 +1,4 @@
import io
import nodes
import node_helpers
import torch
@@ -77,6 +78,7 @@ class ModelSamplingLTXV:
return {"required": { "model": ("MODEL",),
"max_shift": ("FLOAT", {"default": 2.05, "min": 0.0, "max": 100.0, "step":0.01}),
"base_shift": ("FLOAT", {"default": 0.95, "min": 0.0, "max": 100.0, "step":0.01}),
"image_noise_scale": ("FLOAT", {"default": 0.15, "min": 0, "max": 100, "step": 0.01, "tooltip": "Amount of noise to apply on conditioning image latent."})
},
"optional": {"latent": ("LATENT",), }
}
@@ -86,7 +88,7 @@ class ModelSamplingLTXV:
CATEGORY = "advanced/model"
def patch(self, model, max_shift, base_shift, latent=None):
def patch(self, model, max_shift, base_shift, image_noise_scale, latent=None):
m = model.clone()
if latent is None:
@@ -109,6 +111,8 @@ class ModelSamplingLTXV:
model_sampling = ModelSamplingAdvanced(model.model.model_config)
model_sampling.set_parameters(shift=shift)
m.add_object_patch("model_sampling", model_sampling)
m.model_options.setdefault("transformer_options", {})["image_noise_scale"] = image_noise_scale
return (m, )