Steveeeeeeen HF Staff commited on
Commit
001152e
·
verified ·
1 Parent(s): ede9ddf

Update stepaudio2.py

Browse files
Files changed (1) hide show
  1. stepaudio2.py +1 -0
stepaudio2.py CHANGED
@@ -9,6 +9,7 @@ os.environ["HF_TOKEN"] = hf_token
9
  class StepAudio2Base:
10
 
11
  def __init__(self, model_path: str):
 
12
  self.llm_tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True, padding_side="right")
13
  self.llm = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True, torch_dtype=torch.bfloat16).cuda()
14
  self.eos_token_id = self.llm_tokenizer.eos_token_id
 
9
  class StepAudio2Base:
10
 
11
  def __init__(self, model_path: str):
12
+ print(model_path)
13
  self.llm_tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True, padding_side="right")
14
  self.llm = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True, torch_dtype=torch.bfloat16).cuda()
15
  self.eos_token_id = self.llm_tokenizer.eos_token_id