eeuuia commited on
Commit
da0a904
·
verified ·
1 Parent(s): 8097df4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -55,6 +55,16 @@ weight_path = hf_hub_download(
55
  revision=os.getenv("LTXV_REVISION", "8984fa25007f376c1a299016d0957a37a2f797bb")
56
  )
57
 
 
 
 
 
 
 
 
 
 
 
58
  # Carrega o state_dict e aplica no transformer já criado pelo model_index:
59
  state = safe_load(weight_path)
60
  pipeline.transformer.load_state_dict(state, strict=True)
 
55
  revision=os.getenv("LTXV_REVISION", "8984fa25007f376c1a299016d0957a37a2f797bb")
56
  )
57
 
58
+ pipeline = LTXConditionPipeline.from_pretrained(
59
+ "Lightricks/LTX-Video",
60
+ #revision=os.getenv("LTXV_REVISION", "8984fa25007f376c1a299016d0957a37a2f797bb"), # exemplo: SHA atual do repo oficial
61
+ offload_state_dict=False,
62
+ torch_dtype=torch.bfloat16,
63
+ cache_dir=os.getenv("HF_HOME_CACHE"),
64
+ token=os.getenv("HF_TOKEN"),
65
+ )
66
+
67
+
68
  # Carrega o state_dict e aplica no transformer já criado pelo model_index:
69
  state = safe_load(weight_path)
70
  pipeline.transformer.load_state_dict(state, strict=True)