Update README.md
Browse files
README.md
CHANGED
|
@@ -95,9 +95,9 @@ pipe.enable_model_cpu_offload() #for low-vram environments
|
|
| 95 |
|
| 96 |
prompt = "In the video, a miniature cat toy is presented. The cat toy is held in a person's hands. The person then presses on the cat toy, causing a sq41sh squish effect. The person keeps pressing down on the cat toy, further showing the sq41sh squish effect."
|
| 97 |
|
| 98 |
-
image = load_image("https://huggingface.co/
|
| 99 |
|
| 100 |
-
max_area =
|
| 101 |
aspect_ratio = image.height / image.width
|
| 102 |
mod_value = pipe.vae_scale_factor_spatial * pipe.transformer.config.patch_size[1]
|
| 103 |
height = round(np.sqrt(max_area * aspect_ratio)) // mod_value * mod_value
|
|
@@ -111,7 +111,7 @@ output = pipe(
|
|
| 111 |
width=width,
|
| 112 |
num_frames=81,
|
| 113 |
guidance_scale=5.0,
|
| 114 |
-
num_inference_steps=
|
| 115 |
).frames[0]
|
| 116 |
export_to_video(output, "output.mp4", fps=16)
|
| 117 |
```
|
|
|
|
| 95 |
|
| 96 |
prompt = "In the video, a miniature cat toy is presented. The cat toy is held in a person's hands. The person then presses on the cat toy, causing a sq41sh squish effect. The person keeps pressing down on the cat toy, further showing the sq41sh squish effect."
|
| 97 |
|
| 98 |
+
image = load_image("https://huggingface.co/valiantcat/Wan2.1-Fight-LoRA/blob/main/result/test.jpg")
|
| 99 |
|
| 100 |
+
max_area = 512 * 768
|
| 101 |
aspect_ratio = image.height / image.width
|
| 102 |
mod_value = pipe.vae_scale_factor_spatial * pipe.transformer.config.patch_size[1]
|
| 103 |
height = round(np.sqrt(max_area * aspect_ratio)) // mod_value * mod_value
|
|
|
|
| 111 |
width=width,
|
| 112 |
num_frames=81,
|
| 113 |
guidance_scale=5.0,
|
| 114 |
+
num_inference_steps=25
|
| 115 |
).frames[0]
|
| 116 |
export_to_video(output, "output.mp4", fps=16)
|
| 117 |
```
|