eeuuia commited on
Commit
1947e99
·
verified ·
1 Parent(s): 761ae64

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