Spaces:
Sleeping
Sleeping
Update function.py
Browse files- function.py +1 -1
function.py
CHANGED
@@ -2,7 +2,7 @@ from transformers import pipeline
|
|
2 |
import torch
|
3 |
from datasets import load_dataset
|
4 |
import soundfile as sf # 用于保存音频文件
|
5 |
-
from transformers import AutoTokenizer, AutoModelForCausalLM
|
6 |
|
7 |
#tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B")
|
8 |
#model = AutoModelForCausalLM.from_pretrained("deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B")
|
|
|
2 |
import torch
|
3 |
from datasets import load_dataset
|
4 |
import soundfile as sf # 用于保存音频文件
|
5 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM, VitsModel
|
6 |
|
7 |
#tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B")
|
8 |
#model = AutoModelForCausalLM.from_pretrained("deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B")
|