Commit 8939e76b authored by chenpangpang's avatar chenpangpang
Browse files

feat: 增加模型

parent fa8ce810
......@@ -9,4 +9,4 @@ RUN pip install -r requirements.txt
RUN cd llama.cpp && make GGML_CUDA=1 -j
COPY chenyh/$IMAGE/Meta-Llama-3-8B-Instruct-Q4_K_M.gguf /root/$IMAGE/$IMAGE_UPPER/llama.cpp/
COPY chenyh/$IMAGE/tts_models--en--vctk--vits /root/.local/share/tts/
COPY chenyh/$IMAGE/distil-whisper/distil-large-v2 /root/$IMAGE/$IMAGE_UPPER/
#COPY chenyh/$IMAGE/distil-whisper/large-v2 /root/$IMAGE/$IMAGE_UPPER/
......@@ -3,7 +3,7 @@ import os
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
model_list = [
"distil-whisper/distil-large-v2"
"distil-whisper/large-v2"
]
for model_path in model_list:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment