Wan2.1/tests
google-labs-jules[bot] 395fbc40a3 feat: Add --output_format argument for GIF/MP4 output
This commit introduces a new command-line argument `--output_format`
to `generate.py`, allowing users to specify the desired output format
for generated videos. Supported formats are `mp4` (default) and `gif`.

The video saving logic has been updated to handle both formats, using
`imageio` for GIF generation. The `README.md` has been updated to
document this new feature, and a test case has been added to
`tests/test.sh` (though it was adapted for a dry run due to
environmental limitations during testing).
2025-06-25 18:51:51 +00:00
..
README.md init upload 2025-02-25 22:07:47 +08:00
test.sh feat: Add --output_format argument for GIF/MP4 output 2025-06-25 18:51:51 +00:00

Put all your models (Wan2.1-T2V-1.3B, Wan2.1-T2V-14B, Wan2.1-I2V-14B-480P, Wan2.1-I2V-14B-720P) in a folder and specify the max GPU number you want to use.

bash ./test.sh <local model dir> <gpu number>