FYI: Some dependencies are missing.

[im2ex]

Thanks for open sourcing your code :-)

To make example run in fresh conda these additions are necessary:

--- a/pyproject.toml
+++ b/pyproject.toml
@@ -24,7 +24,11 @@ requires-python = ">=3.8"
dependencies = [
"ms-swift==3.3",
"transformers>=4.51.0",

• "peft>=0.15.2"

• "peft>=0.15.2",
• "opencv-python",
• "loguru",
• "timm",
• "flash-attn"
  ]

For the example to run on 24GB VRAM (4090) reduction to batch_num=8 is necessary. (Maybe use as default in quickstart since this is a common VRAM size)

[7-Z-7]

Thank you very much for your suggestion！ The code has been updated.