Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -64,25 +64,25 @@ print("=== Carregando componentes da pipeline... ===")
|
|
| 64 |
vae = AutoModel.from_pretrained(
|
| 65 |
"Lightricks/LTX-Video",
|
| 66 |
subfolder="vae",
|
| 67 |
-
|
| 68 |
cache_dir=CACHE_DIR
|
| 69 |
)
|
| 70 |
text_encoder = AutoModel.from_pretrained(
|
| 71 |
"Lightricks/LTX-Video",
|
| 72 |
subfolder="text_encoder",
|
| 73 |
-
|
| 74 |
cache_dir=CACHE_DIR
|
| 75 |
)
|
| 76 |
scheduler = AutoModel.from_pretrained(
|
| 77 |
"Lightricks/LTX-Video",
|
| 78 |
subfolder="scheduler",
|
| 79 |
-
|
| 80 |
cache_dir=CACHE_DIR
|
| 81 |
)
|
| 82 |
tokenizer = AutoModel.from_pretrained(
|
| 83 |
"Lightricks/LTX-Video",
|
| 84 |
subfolder="tokenizer",
|
| 85 |
-
|
| 86 |
cache_dir=CACHE_DIR
|
| 87 |
)
|
| 88 |
|
|
@@ -94,7 +94,7 @@ if hasattr(scheduler.config, 'use_dynamic_shifting') and scheduler.config.use_dy
|
|
| 94 |
transformer = AutoModel.from_pretrained(
|
| 95 |
"Lightricks/LTX-Video",
|
| 96 |
subfolder="transformer",
|
| 97 |
-
|
| 98 |
cache_dir=CACHE_DIR
|
| 99 |
)
|
| 100 |
transformer.enable_layerwise_casting(
|
|
|
|
| 64 |
vae = AutoModel.from_pretrained(
|
| 65 |
"Lightricks/LTX-Video",
|
| 66 |
subfolder="vae",
|
| 67 |
+
|
| 68 |
cache_dir=CACHE_DIR
|
| 69 |
)
|
| 70 |
text_encoder = AutoModel.from_pretrained(
|
| 71 |
"Lightricks/LTX-Video",
|
| 72 |
subfolder="text_encoder",
|
| 73 |
+
|
| 74 |
cache_dir=CACHE_DIR
|
| 75 |
)
|
| 76 |
scheduler = AutoModel.from_pretrained(
|
| 77 |
"Lightricks/LTX-Video",
|
| 78 |
subfolder="scheduler",
|
| 79 |
+
|
| 80 |
cache_dir=CACHE_DIR
|
| 81 |
)
|
| 82 |
tokenizer = AutoModel.from_pretrained(
|
| 83 |
"Lightricks/LTX-Video",
|
| 84 |
subfolder="tokenizer",
|
| 85 |
+
|
| 86 |
cache_dir=CACHE_DIR
|
| 87 |
)
|
| 88 |
|
|
|
|
| 94 |
transformer = AutoModel.from_pretrained(
|
| 95 |
"Lightricks/LTX-Video",
|
| 96 |
subfolder="transformer",
|
| 97 |
+
|
| 98 |
cache_dir=CACHE_DIR
|
| 99 |
)
|
| 100 |
transformer.enable_layerwise_casting(
|