Carlos s
commited on
Update api/ltx_server.py
Browse files- 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"),
|