Eueuiaa commited on
Commit
e4d0881
·
verified ·
1 Parent(s): b6477f5

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +3 -2
api/ltx_server.py CHANGED
@@ -258,10 +258,10 @@ class LatentConditioningItem:
258
 
259
  def _aduc_prepare_conditioning_patch(
260
  self: "LTXVideoPipeline",
261
- conditioning_items: Optional[List[Union["ConditioningItem", "LatentConditioningItem"]]],
262
  init_latents: torch.Tensor,
263
  num_frames: int,
264
- height: int,
265
  width: int,
266
  vae_per_channel_normalize: bool = False,
267
  generator=None,
@@ -982,6 +982,7 @@ class VideoService:
982
 
983
  second_pass_kwargs = call_kwargs.copy()
984
  second_pass_kwargs.update({
 
985
  "output_type": "latent",
986
  "width": second_pass_width,
987
  "height": second_pass_height,
 
258
 
259
  def _aduc_prepare_conditioning_patch(
260
  self: "LTXVideoPipeline",
261
+ conditioning_itemsl: Optional[List[Union["ConditioningItem", "LatentConditioningItem"]]],
262
  init_latents: torch.Tensor,
263
  num_frames: int,
264
+ height: non_blocking=True,
265
  width: int,
266
  vae_per_channel_normalize: bool = False,
267
  generator=None,
 
982
 
983
  second_pass_kwargs = call_kwargs.copy()
984
  second_pass_kwargs.update({
985
+ "conditioning_items": None,
986
  "output_type": "latent",
987
  "width": second_pass_width,
988
  "height": second_pass_height,