Add FramePack example

[mengniwang95]

User description

Type of Change

example

Description

depends on main branch of auto_round

---

PR Type

Enhancement

---

Description

• Added FramePack example with benchmark script

• Included README with setup instructions and usage examples

• Enhanced script to handle multiple GPU configurations

---

Diagram Walkthrough

flowchart LR
  A["Add run_benchmark.sh"] -- "Handles parameters and runs benchmark" --> B["Add README.md"]
  B -- "Provides setup and usage instructions" --> C["Enhance script for multi-GPU"]

Loading

File Walkthrough

	Relevant files
Enhancement	run_benchmark.sh Add benchmark script                                                                          examples/pytorch/diffusion_model/diffusers/framepack/run_benchmark.sh Added script to initialize parameters and run benchmark Handles different topologies and GPU configurations +128/-0
Documentation	README.md Add README for FramePack example                                                  examples/pytorch/diffusion_model/diffusers/framepack/README.md Added README with environment setup and usage examples +57/-0

---

[PRAgent4INC]

PR Reviewer Guide 🔍

Here are some key observations to aid the review process:

⏱️ Estimated effort to review: 4 🔵🔵🔵🔵⚪

🧪 No relevant tests

🔒 No security concerns identified

⚡ Recommended focus areas for review Error Handling The script does not handle cases where required parameters (e.g., topology, dataset_location) are not provided. It should include checks to ensure all necessary parameters are set before proceeding. function init_params { for var in "$@" do case $var in --topology=*) topology=$(echo $var |cut -f2 -d=) ;; --dataset_location=*) dataset_location=$(echo $var |cut -f2 -d=) ;; --ratio=*) ratio=$(echo $var |cut -f2 -d=) ;; --input_model=*) input_model=$(echo $var |cut -f2 -d=) ;; --limit=*) limit=$(echo $var |cut -f2 -d=) ;; --output_video_path=*) output_video_path=$(echo $var |cut -f2 -d=) ;; --result_path=*) result_path=$(echo $var |cut -f2 -d=) ;; --dimension_list=*) dimension_list=$(echo $var |cut -f2 -d=) ;; *) echo "Error: No such parameter: ${var}" exit 1 ;; esac Variable Initialization Variables like topology, dataset_location, etc., are used without being initialized. Ensure these variables are either initialized with default values or checked for existence before use. function init_params { for var in "$@" do case $var in --topology=*) topology=$(echo $var |cut -f2 -d=) ;; --dataset_location=*) dataset_location=$(echo $var |cut -f2 -d=) ;; --ratio=*) ratio=$(echo $var |cut -f2 -d=) ;; --input_model=*) input_model=$(echo $var |cut -f2 -d=) ;; --limit=*) limit=$(echo $var |cut -f2 -d=) ;; --output_video_path=*) output_video_path=$(echo $var |cut -f2 -d=) ;; --result_path=*) result_path=$(echo $var |cut -f2 -d=) ;; --dimension_list=*) dimension_list=$(echo $var |cut -f2 -d=) ;; *) echo "Error: No such parameter: ${var}" exit 1 ;; esac Parallel Execution The script runs multiple instances of main.py in parallel using &. Ensure that there are no race conditions or shared resources that could cause issues when running in parallel. for ((i=0; i<count; i++)); do export CUDA_VISIBLE_DEVICES=${gpu_ids[i]} python3 main.py \ --output_video_path ${output_video_path} \ --dataset_location ${dataset_location} \ --ratio ${ratio} \ --limit ${limit} \ --dimension_list ${dimensions[i]} \ ${extra_cmd} & program_pid+=($!) echo "Start (PID: ${program_pid[-1]}, GPU: ${i})" done wait "${program_pid[@]}" else

[PRAgent4INC]

Failed to generate code suggestions for PR

[chensuyue]

Can we freeze to a tag or commit?

[mengniwang95]

done

[chensuyue]

use set -ex, so script will exit with the first error.

[chensuyue]

add gdown, used by sh vbench2_beta_i2v/download_data.sh

[chensuyue]

let's freeze the deps versions.