Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -53,6 +53,9 @@ transformer.enable_layerwise_casting(
|
|
| 53 |
storage_dtype=torch.float8_e4m3fn, compute_dtype=dtype
|
| 54 |
)
|
| 55 |
|
|
|
|
|
|
|
|
|
|
| 56 |
# 3. Carregar a pipeline completa, injetando o Transformer já otimizado
|
| 57 |
print(f"Carregando a arquitetura da pipeline de {base_model_repo}...")
|
| 58 |
pipeline = LTXConditionPipeline.from_pretrained(
|
|
|
|
| 53 |
storage_dtype=torch.float8_e4m3fn, compute_dtype=dtype
|
| 54 |
)
|
| 55 |
|
| 56 |
+
print("Desativando 'dynamic shifting' para compatibilidade com a pipeline.")
|
| 57 |
+
transformer.config.use_dynamic_shifting = False
|
| 58 |
+
|
| 59 |
# 3. Carregar a pipeline completa, injetando o Transformer já otimizado
|
| 60 |
print(f"Carregando a arquitetura da pipeline de {base_model_repo}...")
|
| 61 |
pipeline = LTXConditionPipeline.from_pretrained(
|