423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-02 19:52:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

[webui] fix input args (#8162)

Browse Source
This commit is contained in:
hoshi-hiyouga 2025-05-27 02:05:54 +08:00 committed by GitHub
parent f3a1dc8483
commit 09436c1f45
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/llamafactory/webui/runner.py
View File

@ -29,6 +29,7 @@ from .common import (
DEFAULT_CACHE_DIR, DEFAULT_CACHE_DIR,
DEFAULT_CONFIG_DIR, DEFAULT_CONFIG_DIR,
abort_process, abort_process,
calculate_pixels,
gen_cmd, gen_cmd,
get_save_dir, get_save_dir,
load_args, load_args,
@ -241,10 +242,10 @@ class Runner:
args["freeze_vision_tower"] = get("train.freeze_vision_tower") args["freeze_vision_tower"] = get("train.freeze_vision_tower")
args["freeze_multi_modal_projector"] = get("train.freeze_multi_modal_projector") args["freeze_multi_modal_projector"] = get("train.freeze_multi_modal_projector")
args["freeze_language_model"] = get("train.freeze_language_model") args["freeze_language_model"] = get("train.freeze_language_model")
args["image_max_pixels"] = get("train.image_max_pixels") args["image_max_pixels"] = calculate_pixels(get("train.image_max_pixels"))
args["image_min_pixels"] = get("train.image_min_pixels") args["image_min_pixels"] = calculate_pixels(get("train.image_min_pixels"))
args["video_max_pixels"] = get("train.video_max_pixels") args["video_max_pixels"] = calculate_pixels(get("train.video_max_pixels"))
args["video_min_pixels"] = get("train.video_min_pixels") args["video_min_pixels"] = calculate_pixels(get("train.video_min_pixels"))
# galore config # galore config
if args["use_galore"]: if args["use_galore"]: