Update handler.py
Browse files- 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 |
-
|
470 |
-
|
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 = {
|