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