fix: Add seed input to AIIA_DittoSampler node definition to resolve NameError
This commit is contained in:
+2
-1
@@ -428,6 +428,7 @@ class AIIA_DittoSampler:
|
||||
"relax_on_silence": ("BOOLEAN", {"default": True, "label_on": "Relax Face on Silence", "label_off": "Disabled"}),
|
||||
"ref_threshold": ("FLOAT", {"default": 0.005, "min": 0.0, "max": 1.0, "step": 0.001}),
|
||||
"blink_mode": (["Random (Normal)", "Fast", "Slow", "None"], {"default": "Random (Normal)"}),
|
||||
"seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}),
|
||||
}
|
||||
}
|
||||
|
||||
@@ -436,7 +437,7 @@ class AIIA_DittoSampler:
|
||||
FUNCTION = "generate"
|
||||
CATEGORY = "AIIA/Ditto"
|
||||
|
||||
def generate(self, pipe, ref_image, audio, sampling_steps, fps, crop_scale, emo, drive_eye, chk_eye_blink, smo_k_d, hd_rot_p, hd_rot_y, hd_rot_r, mouth_amp, relax_on_silence, ref_threshold, blink_mode):
|
||||
def generate(self, pipe, ref_image, audio, sampling_steps, fps, crop_scale, emo, drive_eye, chk_eye_blink, smo_k_d, hd_rot_p, hd_rot_y, hd_rot_r, mouth_amp, relax_on_silence, ref_threshold, blink_mode, seed):
|
||||
# pipe is the dict we returned in Loader
|
||||
master_sdk = pipe["sdk"]
|
||||
cfg_pkl = pipe["cfg_pkl"]
|
||||
|
||||
Reference in New Issue
Block a user