Spaces:
Running
Running
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)
|