fffiloni commited on
Commit
df253f2
1 Parent(s): 51df367

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -61,7 +61,7 @@ def infer(image_path, prompt, orbit_type, progress=gr.Progress(track_tqdm=True))
61
 
62
  lora_path = "checkpoints/"
63
  weight_name = "orbit_left_lora_weights.safetensors" if orbit_type == "Left" else "orbit_up_lora_weights.safetensors"
64
- lora_rank = 128
65
  adapter_timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
66
 
67
  # Load LoRA weights on CPU
@@ -81,7 +81,7 @@ def infer(image_path, prompt, orbit_type, progress=gr.Progress(track_tqdm=True))
81
  video = pipe(
82
  image,
83
  prompt,
84
- num_inference_steps=25,
85
  guidance_scale=7.0,
86
  use_dynamic_cfg=True,
87
  generator=torch.Generator(device="cpu").manual_seed(seed)
@@ -156,4 +156,4 @@ with gr.Blocks(analytics_enabled=False) as demo:
156
  outputs=[video_out]
157
  )
158
 
159
- demo.queue().launch(show_error=True, show_api=False)
 
61
 
62
  lora_path = "checkpoints/"
63
  weight_name = "orbit_left_lora_weights.safetensors" if orbit_type == "Left" else "orbit_up_lora_weights.safetensors"
64
+ lora_rank = 256
65
  adapter_timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
66
 
67
  # Load LoRA weights on CPU
 
81
  video = pipe(
82
  image,
83
  prompt,
84
+ num_inference_steps=50,
85
  guidance_scale=7.0,
86
  use_dynamic_cfg=True,
87
  generator=torch.Generator(device="cpu").manual_seed(seed)
 
156
  outputs=[video_out]
157
  )
158
 
159
+ demo.queue().launch(show_error=True, show_api=False, ssr_mode=False)