eeuuia commited on
Commit
073093f
·
verified ·
1 Parent(s): 2046b2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -96,12 +96,19 @@ transformer.enable_layerwise_casting(
96
  )
97
 
98
 
99
- # 4. Montar a pipeline principal
100
- print("Montando a LTXConditionPipeline...")
101
- pipeline = LTXConditionPipeline(
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