Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -96,12 +96,19 @@ transformer.enable_layerwise_casting(
|
|
| 96 |
)
|
| 97 |
|
| 98 |
|
| 99 |
-
|
| 100 |
-
|
| 101 |
-
|
| 102 |
vae=vae, text_encoder=text_encoder, tokenizer=tokenizer,
|
| 103 |
-
scheduler=scheduler, transformer=transformer
|
|
|
|
|
|
|
|
|
|
| 104 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 105 |
pipeline.to(device)
|
| 106 |
pipeline.vae.enable_tiling()
|
| 107 |
|
|
|
|
| 96 |
)
|
| 97 |
|
| 98 |
|
| 99 |
+
pipeline = LTXConditionPipeline.from_pretrained(
|
| 100 |
+
"Lightricks/LTX-Video-0.9.8-13B-distilled",
|
| 101 |
+
offload_state_dict=False,
|
| 102 |
vae=vae, text_encoder=text_encoder, tokenizer=tokenizer,
|
| 103 |
+
scheduler=scheduler, transformer=transformer,
|
| 104 |
+
torch_dtype=torch.bfloat16,
|
| 105 |
+
cache_dir=os.getenv("HF_HOME_CACHE"),
|
| 106 |
+
token=os.getenv("HF_TOKEN"),
|
| 107 |
)
|
| 108 |
+
|
| 109 |
+
|
| 110 |
+
# 4. Montar a pipeline principal
|
| 111 |
+
|
| 112 |
pipeline.to(device)
|
| 113 |
pipeline.vae.enable_tiling()
|
| 114 |
|