diff --git a/xinference/model/llm/pytorch/rwkv.py b/xinference/model/llm/pytorch/rwkv.py index af3d1a5a5c..7a43aef873 100644 --- a/xinference/model/llm/pytorch/rwkv.py +++ b/xinference/model/llm/pytorch/rwkv.py @@ -39,7 +39,7 @@ def __init__( def _load_model(self, kwargs: dict): try: - from transformers import RwkvForCausalLM, AutoTokenizer + from transformers import AutoTokenizer, RwkvForCausalLM except ImportError: error_message = "Failed to import module 'transformers'" installation_guide = [