Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
)
|
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)
|