Spaces:
Paused
Paused
Update api/ltx_server_refactored_complete.py
Browse files
api/ltx_server_refactored_complete.py
CHANGED
|
@@ -295,13 +295,16 @@ class VideoService:
|
|
| 295 |
}
|
| 296 |
|
| 297 |
logging.info(f"\n[Info] pipeline_kwargs:\n {pipeline_kwargs}\n\n")
|
| 298 |
-
self._log_conditioning_items(kwargs['conditioning_items'])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 299 |
|
| 300 |
with torch.autocast(device_type=self.main_device.type, dtype=self.runtime_autocast_dtype, enabled="cuda" in self.main_device.type):
|
| 301 |
-
latents_raw = self.pipeline(
|
| 302 |
-
"conditioning_items": kwargs['conditioning_items']
|
| 303 |
-
**pipeline_kwargs
|
| 304 |
-
).images
|
| 305 |
|
| 306 |
return latents_raw.to(self.main_device)
|
| 307 |
|
|
|
|
| 295 |
}
|
| 296 |
|
| 297 |
logging.info(f"\n[Info] pipeline_kwargs:\n {pipeline_kwargs}\n\n")
|
| 298 |
+
self._log_conditioning_items(kwargs.get['conditioning_items'])
|
| 299 |
+
|
| 300 |
+
pipeline_kwargs = {
|
| 301 |
+
"conditioning_items": kwargs['conditioning_items'],
|
| 302 |
+
**pipeline_kwargs
|
| 303 |
+
}
|
| 304 |
+
|
| 305 |
|
| 306 |
with torch.autocast(device_type=self.main_device.type, dtype=self.runtime_autocast_dtype, enabled="cuda" in self.main_device.type):
|
| 307 |
+
latents_raw = self.pipeline(**pipeline_kwargs).images
|
|
|
|
|
|
|
|
|
|
| 308 |
|
| 309 |
return latents_raw.to(self.main_device)
|
| 310 |
|