diff --git a/src/model_download/index.py b/src/model_download/index.py index 0d2cc7a..3575ee7 100644 --- a/src/model_download/index.py +++ b/src/model_download/index.py @@ -39,7 +39,8 @@ def handler(event, context): # using latest ollama print('[INFO] Downloading and installing the latest ollama. ') - latest_ollama = api.list_model_revisions(model_id='modelscope/ollama-linux')[0] + # latest_ollama = api.list_model_revisions(model_id='modelscope/ollama-linux')[0] + latest_ollama = 'v0.3.6' os.system(f'modelscope download --model=modelscope/ollama-linux --local_dir {cache_dir}/ollama-linux --revision {latest_ollama}') print(f'[INFO] Downloading model file. ')