Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
Jintao-Huang committed Dec 31, 2024
1 parent f21fd3f commit d8655e9
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions tests/llm/test_custom.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,9 @@
# Copyright (c) Alibaba, Inc. and its affiliates.
import unittest
from typing import Any, Dict, Optional

import torch

from swift.llm import (DatasetMeta, InferRequest, Model, ModelGroup, ModelMeta, PtEngine, RequestConfig,
ResponsePreprocessor, TemplateMeta, get_model_tokenizer_with_flash_attn, load_dataset,
register_dataset, register_model, register_template)
Expand Down

0 comments on commit d8655e9

Please sign in to comment.