423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2026-03-04 02:35:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

[v1] add cli sampler (#9721)

Browse Source
This commit is contained in:
Yaowei Zheng
2026-01-06 23:31:27 +08:00
committed by GitHub
parent e944dc442c
commit ea0b4e2466
45 changed files with 1091 additions and 505 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/llamafactory/v1/config/arg_utils.py
View File

@@ -17,7 +17,7 @@
import json
from enum import Enum, unique
from enum import StrEnum, unique
class PluginConfig(dict):
@@ -36,7 +36,7 @@ PluginArgument = PluginConfig | dict | str | None
@unique
class ModelClass(str, Enum):
class ModelClass(StrEnum):
"""Auto class for model config."""
LLM = "llm"
@@ -45,7 +45,7 @@ class ModelClass(str, Enum):
@unique
class SampleBackend(str, Enum):
class SampleBackend(StrEnum):
HF = "hf"
VLLM = "vllm"