Spaces:
Running
on
Zero
Running
on
Zero
Update stepaudio2.py
Browse files- 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
|