Update api/ltx_server.py
Browse files- 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 |
-
|
| 262 |
init_latents: torch.Tensor,
|
| 263 |
num_frames: int,
|
| 264 |
-
height:
|
| 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,
|