jbilcke-hf HF staff commited on
Commit
306dec1
·
verified ·
1 Parent(s): 1b3852a

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +5 -5
handler.py CHANGED
@@ -265,8 +265,8 @@ class EndpointHandler:
265
  #self.image_to_video.enable_model_cpu_offload()
266
 
267
  # Inject enhance-a-video functionality
268
- inject_enhance_for_ltx(self.text_to_video.transformer)
269
- inject_enhance_for_ltx(self.image_to_video.transformer)
270
 
271
  # Initialize Varnish for post-processing
272
  self.varnish = Varnish(
@@ -465,9 +465,9 @@ class EndpointHandler:
465
  generator = generator.manual_seed(config.seed)
466
 
467
  # Configure enhance-a-video
468
- if config.enable_enhance_a_video:
469
- enable_enhance()
470
- set_enhance_weight(config.enhance_a_video_weight)
471
 
472
  # Prepare generation parameters for the video model (we omit params that are destined to Varnish, or things like the seed which is set externally)
473
  generation_kwargs = {
 
265
  #self.image_to_video.enable_model_cpu_offload()
266
 
267
  # Inject enhance-a-video functionality
268
+ #inject_enhance_for_ltx(self.text_to_video.transformer)
269
+ #inject_enhance_for_ltx(self.image_to_video.transformer)
270
 
271
  # Initialize Varnish for post-processing
272
  self.varnish = Varnish(
 
465
  generator = generator.manual_seed(config.seed)
466
 
467
  # Configure enhance-a-video
468
+ #if config.enable_enhance_a_video:
469
+ # enable_enhance()
470
+ # set_enhance_weight(config.enhance_a_video_weight)
471
 
472
  # Prepare generation parameters for the video model (we omit params that are destined to Varnish, or things like the seed which is set externally)
473
  generation_kwargs = {