matt/ollama37
1
0
Fork 0
mirror of https://github.com/dogkeeper886/ollama37.git synced 2025-12-12 00:37:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

use all caps for constants

Browse Source
This commit is contained in:
Michael Yang
2023-06-30 10:35:26 -07:00
parent 4d0eb7639a
commit 07d8d56177
3 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ollama/engine.py
View File

@@ -7,7 +7,7 @@ from llama_cpp import Llama
from ctransformers import AutoModelForCausalLM
import ollama.prompt
from ollama.model import models_home
from ollama.model import MODELS_CACHE_PATH
@contextmanager
@@ -30,7 +30,7 @@ def load(model_name, models={}):
if not models.get(model_name, None):
model_path = path.expanduser(model_name)
if not path.exists(model_path):
model_path = path.join(models_home, model_name + ".bin")
model_path = path.join(MODELS_CACHE_PATH, model_name + ".bin")
runners = {
model_type: cls