Update stepaudio2.py

stepaudio2.py

@@ -10,8 +10,8 @@ class StepAudio2Base:
 
     def __init__(self, model_path: str):
         print(model_path)
-        self.llm_tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True, padding_side="right")
-        self.llm = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True, torch_dtype=torch.bfloat16).cuda()
+        self.llm_tokenizer = AutoTokenizer.from_pretrained("stepfun-ai/Step-Audio-2-mini", trust_remote_code=True, padding_side="right")
+        self.llm = AutoModelForCausalLM.from_pretrained("stepfun-ai/Step-Audio-2-mini", trust_remote_code=True, torch_dtype=torch.bfloat16).cuda()
         self.eos_token_id = self.llm_tokenizer.eos_token_id
 
     def __call__(self, messages: list, **kwargs):