Carlos s commited on
Commit
09305c5
·
verified ·
1 Parent(s): 85d172d

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +6 -0
api/ltx_server.py CHANGED
@@ -718,6 +718,8 @@ class VideoService:
718
  "num_inference_steps": first_pass_config.get("num_inference_steps"),
719
  "skip_final_inference_steps": first_pass_config.get("skip_final_inference_steps"),
720
  "cfg_star_rescale": first_pass_config.get("cfg_star_rescale"),
 
 
721
  #"guidance_scale": float(guidance_scale),
722
  #"stg_scale": first_pass_config.get("stg_scale"),
723
  #"rescaling_scale": first_pass_config.get("rescaling_scale"),
@@ -749,6 +751,8 @@ class VideoService:
749
 
750
  single_pass_kwargs = call_kwargs.copy()
751
  first_pass_config = self.config.get("first_pass", {}).copy()
 
 
752
 
753
  single_pass_kwargs.update(
754
  {
@@ -762,6 +766,8 @@ class VideoService:
762
  "num_inference_steps": second_pass.get("num_inference_steps"),
763
  "skip_final_inference_steps": second_pass.get("skip_final_inference_steps"),
764
  "cfg_star_rescale": second_pass.get("cfg_star_rescale"),
 
 
765
  #"guidance_scale": float(guidance_scale),
766
  #"stg_scale": first_pass_config.get("stg_scale"),
767
  #"rescaling_scale": first_pass_config.get("rescaling_scale"),
 
718
  "num_inference_steps": first_pass_config.get("num_inference_steps"),
719
  "skip_final_inference_steps": first_pass_config.get("skip_final_inference_steps"),
720
  "cfg_star_rescale": first_pass_config.get("cfg_star_rescale"),
721
+ "downscale_factor": self.config["downscale_factor"],
722
+
723
  #"guidance_scale": float(guidance_scale),
724
  #"stg_scale": first_pass_config.get("stg_scale"),
725
  #"rescaling_scale": first_pass_config.get("rescaling_scale"),
 
751
 
752
  single_pass_kwargs = call_kwargs.copy()
753
  first_pass_config = self.config.get("first_pass", {}).copy()
754
+ second_pass_args = self.config.get("second_pass", {}).copy()
755
+ second_pass = self.config.get("second_pass", {}).copy()
756
 
757
  single_pass_kwargs.update(
758
  {
 
766
  "num_inference_steps": second_pass.get("num_inference_steps"),
767
  "skip_final_inference_steps": second_pass.get("skip_final_inference_steps"),
768
  "cfg_star_rescale": second_pass.get("cfg_star_rescale"),
769
+ "downscale_factor": self.config["downscale_factor"],
770
+ "second_pass": second_pass_args,
771
  #"guidance_scale": float(guidance_scale),
772
  #"stg_scale": first_pass_config.get("stg_scale"),
773
  #"rescaling_scale": first_pass_config.get("rescaling_scale"),