Update api/ltx_server.py
Browse files- api/ltx_server.py +6 -6
api/ltx_server.py
CHANGED
|
@@ -331,7 +331,7 @@ class VideoService:
|
|
| 331 |
self.pipeline, self.latent_upsampler = self._load_models()
|
| 332 |
print(f"[DEBUG] Pipeline e Upsampler carregados. Upsampler ativo? {bool(self.latent_upsampler)}")
|
| 333 |
|
| 334 |
-
self._apply_ltx_pipeline_patches()
|
| 335 |
|
| 336 |
print(f"[DEBUG] Movendo modelos para {self.device}...")
|
| 337 |
self.pipeline.to(self.device)
|
|
@@ -949,13 +949,13 @@ class VideoService:
|
|
| 949 |
#latents_parts_up = self._dividir_latentes_por_tamanho(latents_cpu_up,15,0)
|
| 950 |
|
| 951 |
|
| 952 |
-
|
| 953 |
-
|
| 954 |
-
|
| 955 |
|
| 956 |
-
|
| 957 |
|
| 958 |
-
latents_parts_up = [latents_cpu_up]
|
| 959 |
|
| 960 |
|
| 961 |
for latents in latents_parts_up:
|
|
|
|
| 331 |
self.pipeline, self.latent_upsampler = self._load_models()
|
| 332 |
print(f"[DEBUG] Pipeline e Upsampler carregados. Upsampler ativo? {bool(self.latent_upsampler)}")
|
| 333 |
|
| 334 |
+
#self._apply_ltx_pipeline_patches()
|
| 335 |
|
| 336 |
print(f"[DEBUG] Movendo modelos para {self.device}...")
|
| 337 |
self.pipeline.to(self.device)
|
|
|
|
| 949 |
#latents_parts_up = self._dividir_latentes_por_tamanho(latents_cpu_up,15,0)
|
| 950 |
|
| 951 |
|
| 952 |
+
lat_aup, lat_bup = self._dividir_latentes(latents_cpu_up)
|
| 953 |
+
print(f"[DEBUG] Partição Aup: {tuple(lat_aup.shape)}")
|
| 954 |
+
print(f"[DEBUG] Partição Bup: {tuple(lat_bup.shape)}")
|
| 955 |
|
| 956 |
+
latents_parts_up = [lat_aup, lat_bup]
|
| 957 |
|
| 958 |
+
#latents_parts_up = [latents_cpu_up]
|
| 959 |
|
| 960 |
|
| 961 |
for latents in latents_parts_up:
|