Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -90,21 +90,7 @@ if True:
|
|
| 90 |
# 3. Define a precisão dos modelos (ainda na CPU, será aplicado na GPU depois)
|
| 91 |
|
| 92 |
# 4. Monta o objeto do Pipeline com os componentes carregados
|
| 93 |
-
|
| 94 |
-
submodel_dict = {
|
| 95 |
-
"transformer": transformer,
|
| 96 |
-
"patchifier": patchifier,
|
| 97 |
-
"text_encoder": text_encoder,
|
| 98 |
-
"tokenizer": tokenizer,
|
| 99 |
-
"scheduler": scheduler,
|
| 100 |
-
"vae": vae,
|
| 101 |
-
#"allowed_inference_steps": allowed_inference_steps,
|
| 102 |
-
# Os prompt enhancers são opcionais e não são carregados por padrão para economizar memória
|
| 103 |
-
"prompt_enhancer_image_caption_model": None,
|
| 104 |
-
"prompt_enhancer_image_caption_processor": None,
|
| 105 |
-
"prompt_enhancer_llm_model": None,
|
| 106 |
-
"prompt_enhancer_llm_tokenizer": None,
|
| 107 |
-
}
|
| 108 |
pipeline = LTXConditionPipeline.from_pretrained(weight_path, cache_dir=os.getenv("HF_HOME_CACHE"), torch_dtype=dtype)
|
| 109 |
|
| 110 |
|
|
|
|
| 90 |
# 3. Define a precisão dos modelos (ainda na CPU, será aplicado na GPU depois)
|
| 91 |
|
| 92 |
# 4. Monta o objeto do Pipeline com os componentes carregados
|
| 93 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 94 |
pipeline = LTXConditionPipeline.from_pretrained(weight_path, cache_dir=os.getenv("HF_HOME_CACHE"), torch_dtype=dtype)
|
| 95 |
|
| 96 |
|