Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
11 |
####################
|
12 |
# 全局加载模型
|
13 |
####################
|
14 |
-
llasa_3b = "HKUSTAudio/Llasa-1B-multi-speakers-genshin"
|
15 |
print("Loading tokenizer & model ...")
|
16 |
tokenizer = AutoTokenizer.from_pretrained(llasa_3b)
|
17 |
model = AutoModelForCausalLM.from_pretrained(llasa_3b)
|
|
|
11 |
####################
|
12 |
# 全局加载模型
|
13 |
####################
|
14 |
+
llasa_3b = "HKUSTAudio/Llasa-1B-multi-speakers-genshin-zh-en-ja-ko"
|
15 |
print("Loading tokenizer & model ...")
|
16 |
tokenizer = AutoTokenizer.from_pretrained(llasa_3b)
|
17 |
model = AutoModelForCausalLM.from_pretrained(llasa_3b)
|