mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2026-05-05 15:48:54 +08:00
[model] support MiniCPM-o-4.5 (#10163)
Co-authored-by: Yaowei Zheng <hiyouga@buaa.edu.cn>
This commit is contained in:
@@ -27,7 +27,7 @@ Including:
|
||||
import os
|
||||
from collections.abc import Callable
|
||||
from contextlib import contextmanager
|
||||
from enum import Enum, unique
|
||||
from enum import StrEnum, unique
|
||||
from functools import lru_cache, wraps
|
||||
from typing import Optional
|
||||
|
||||
@@ -39,7 +39,7 @@ from ..utils.types import ProcessGroup, Tensor, TensorLike
|
||||
|
||||
|
||||
@unique
|
||||
class DeviceType(str, Enum):
|
||||
class DeviceType(StrEnum):
|
||||
CPU = "cpu"
|
||||
CUDA = "cuda"
|
||||
META = "meta"
|
||||
@@ -49,7 +49,7 @@ class DeviceType(str, Enum):
|
||||
|
||||
|
||||
@unique
|
||||
class ReduceOp(str, Enum):
|
||||
class ReduceOp(StrEnum):
|
||||
SUM = "sum"
|
||||
MEAN = "mean"
|
||||
MAX = "max"
|
||||
|
||||
Reference in New Issue
Block a user