fix timesteps

lldacing · Jan 20, 2025 · 05da865 · 05da865
1 parent b3e2ebf
commit 05da865
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/PulidFluxHook.py b/PulidFluxHook.py
@@ -65,7 +65,7 @@ def add(self, callback, **kwargs):
     def __call__(self, input_args, extra_options):
         transformer_options = extra_options["transformer_options"]
         pulid_temp_attrs = transformer_options.get(PatchKeys.pulid_patch_key_attrs, {})
-        sigma = pulid_temp_attrs["timesteps"].detach().cpu().item()
+        sigma = pulid_temp_attrs["timesteps"][0].detach().cpu().item()
         out = extra_options["original_block"](input_args)
         img = out['img']
         temp_img = img