Spaces:
Paused
Paused
Update app.py
Browse files
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)
|