mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2026-04-28 02:39:03 +08:00
[v1] add accelerator (#9607)
This commit is contained in:
@@ -14,12 +14,8 @@
|
||||
|
||||
|
||||
from dataclasses import dataclass, field
|
||||
from enum import Enum
|
||||
|
||||
|
||||
class SampleBackend(Enum):
|
||||
HF = "hf"
|
||||
VLLM = "vllm"
|
||||
from .arg_utils import SampleBackend
|
||||
|
||||
|
||||
@dataclass
|
||||
|
||||
Reference in New Issue
Block a user