Vicente Alvarez commited on
Commit
fdf8bb8
·
1 Parent(s): 01f47a1

Fix: use upsampler 1.1, default 896x512

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -113,7 +113,7 @@ print("=" * 80)
113
 
114
  checkpoint_path = hf_hub_download(repo_id=CHECKPOINT_REPO, filename="sulphur_dev_bf16.safetensors")
115
  distilled_lora_path = hf_hub_download(repo_id=DISTILL_LORA_REPO, filename="distill_loras/ltx-2.3-22b-distilled-lora-1.1_fro90_ceil72_condsafe.safetensors")
116
- spatial_upsampler_path = hf_hub_download(repo_id=LTX_MODEL_REPO, filename="ltx-2.3-spatial-upscaler-x2-1.0.safetensors")
117
  gemma_root = snapshot_download(repo_id=GEMMA_REPO)
118
 
119
  print(f"Checkpoint: {checkpoint_path}")
@@ -221,8 +221,8 @@ def generate_video(
221
  enhance_prompt: bool = True,
222
  seed: int = 42,
223
  randomize_seed: bool = True,
224
- height: int = 1024,
225
- width: int = 1536,
226
  negative_prompt: str = DEFAULT_NEGATIVE_PROMPT,
227
  progress=gr.Progress(track_tqdm=True),
228
  ):
@@ -353,8 +353,8 @@ with gr.Blocks(title="Element-16 Video") as demo:
353
  seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, value=10, step=1)
354
  randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
355
  with gr.Row():
356
- width = gr.Number(label="Width", value=1536, precision=0)
357
- height = gr.Number(label="Height", value=1024, precision=0)
358
  with gr.Row():
359
  enhance_prompt = gr.Checkbox(label="Enhance Prompt", value=False)
360
  high_res = gr.Checkbox(label="High Resolution", value=True)
 
113
 
114
  checkpoint_path = hf_hub_download(repo_id=CHECKPOINT_REPO, filename="sulphur_dev_bf16.safetensors")
115
  distilled_lora_path = hf_hub_download(repo_id=DISTILL_LORA_REPO, filename="distill_loras/ltx-2.3-22b-distilled-lora-1.1_fro90_ceil72_condsafe.safetensors")
116
+ spatial_upsampler_path = hf_hub_download(repo_id=LTX_MODEL_REPO, filename="ltx-2.3-spatial-upscaler-x2-1.1.safetensors")
117
  gemma_root = snapshot_download(repo_id=GEMMA_REPO)
118
 
119
  print(f"Checkpoint: {checkpoint_path}")
 
221
  enhance_prompt: bool = True,
222
  seed: int = 42,
223
  randomize_seed: bool = True,
224
+ height: int = 512,
225
+ width: int = 896,
226
  negative_prompt: str = DEFAULT_NEGATIVE_PROMPT,
227
  progress=gr.Progress(track_tqdm=True),
228
  ):
 
353
  seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, value=10, step=1)
354
  randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
355
  with gr.Row():
356
+ width = gr.Number(label="Width", value=896, precision=0)
357
+ height = gr.Number(label="Height", value=512, precision=0)
358
  with gr.Row():
359
  enhance_prompt = gr.Checkbox(label="Enhance Prompt", value=False)
360
  high_res = gr.Checkbox(label="High Resolution", value=True)