multimodalart HF Staff commited on
Commit
2a3fbc3
·
verified ·
1 Parent(s): c942f44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -94,14 +94,14 @@ pipe = create_ltx_video_pipeline(
94
  sampler=PIPELINE_CONFIG_YAML["sampler"], # "from_checkpoint" or specific sampler
95
  device=DEVICE,
96
  enhance_prompt=False, # Assuming Gradio controls this, or set based on YAML later
97
- ).to(torch.bfloat16)
98
 
99
  # Create Latent Upsampler
100
  latent_upsampler = create_latent_upsampler(
101
  latent_upsampler_model_path=spatial_upsampler_path,
102
  device=DEVICE
103
  )
104
- latent_upsampler = latent_upsampler.to(torch.bfloat16)
105
 
106
 
107
  # Multi-scale pipeline (wrapper)
 
94
  sampler=PIPELINE_CONFIG_YAML["sampler"], # "from_checkpoint" or specific sampler
95
  device=DEVICE,
96
  enhance_prompt=False, # Assuming Gradio controls this, or set based on YAML later
97
+ )#.to(torch.bfloat16)
98
 
99
  # Create Latent Upsampler
100
  latent_upsampler = create_latent_upsampler(
101
  latent_upsampler_model_path=spatial_upsampler_path,
102
  device=DEVICE
103
  )
104
+ #latent_upsampler = latent_upsampler.to(torch.bfloat16)
105
 
106
 
107
  # Multi-scale pipeline (wrapper)