chore: 修改类名,优化导入顺序

This commit is contained in:
2026-04-05 22:27:57 +08:00
parent 9b22b1651e
commit 39766aa1dc
21 changed files with 40 additions and 47 deletions
+2 -2
View File
@@ -9,7 +9,7 @@ import torch
from torch import Tensor
from astrai.model.automodel import AutoModel
from astrai.tokenize.tokenizer import TextTokenizer
from astrai.tokenize import AutoTokenizer
class TaskStatus:
@@ -101,7 +101,7 @@ class InferenceScheduler:
def __init__(
self,
model: AutoModel,
tokenizer: TextTokenizer,
tokenizer: AutoTokenizer,
max_batch_size: int = 16,
max_seq_len: Optional[int] = None,
device: str = "cuda",