eeuuia commited on
Commit
29dad4c
·
verified ·
1 Parent(s): 8c7b9f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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(