py/OmniLottie
1
0
Fork 0
mirror of https://github.com/OpenVGLab/OmniLottie.git synced 2026-09-16 23:26:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support HuggingFace format for OmniLottie.

Browse Source
This commit is contained in:
Yiying Yang
2026-03-20 18:01:49 +08:00
committed by GitHub
parent 02917fe1c4
commit 59fbbf9dc5
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app_hf.py
+2 -2
View File
@@ -6,7 +6,7 @@ supporting automatic downloading from Hugging Face Hub.
Environment Variables: Environment Variables:
MODEL_PATH: Model path or HF Hub ID (default: "OmniLottie/OmniLottie") MODEL_PATH: Model path or HF Hub ID (default: "OmniLottie/OmniLottie")
PROCESSOR_PATH: Processor path (default: "/mnt/jfs-test/Qwen2.5-VL-3B-Instruct") PROCESSOR_PATH: Processor path (default: "Qwen/Qwen2.5-VL-3B-Instruct")
Usage: Usage:
# Using HF Hub model (automatic download) # Using HF Hub model (automatic download)
@@ -84,7 +84,7 @@ def load_model_once():
model = model.to(device).eval() model = model.to(device).eval()
# Load processor # Load processor
processor_path = os.environ.get("PROCESSOR_PATH", "/mnt/jfs-test/Qwen2.5-VL-3B-Instruct") processor_path = os.environ.get("PROCESSOR_PATH", "Qwen/Qwen2.5-VL-3B-Instruct")
processor = AutoProcessor.from_pretrained( processor = AutoProcessor.from_pretrained(
processor_path, processor_path,
padding_side="left", padding_side="left",