baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge pull request #23 from jmorganca/fix-imports

Browse source 
fix imports
This commit is contained in:
Michael Yang 2023-06-29 16:15:06 -07:00 committed by GitHub
commit 42047b15de
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
ollama/engine.py
View file

@ -1,5 +1,6 @@
from os import path, dup, dup2, devnull
import os
import sys
from os import path
from contextlib import contextmanager
from llama_cpp import Llama as LLM
@ -9,12 +10,12 @@ import ollama.prompt
@contextmanager
def suppress_stderr():
stderr = dup(sys.stderr.fileno())
with open(devnull, "w") as devnull:
dup2(devnull.fileno(), sys.stderr.fileno())
stderr = os.dup(sys.stderr.fileno())
with open(os.devnull, "w") as devnull:
os.dup2(devnull.fileno(), sys.stderr.fileno())
yield
dup2(stderr, sys.stderr.fileno())
os.dup2(stderr, sys.stderr.fileno())
def generate(model, prompt, models_home=".", llms={}, *args, **kwargs):