highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-09-01 20:31:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix imports

Browse Source
This commit is contained in:
Michael Yang
2023-06-29 16:13:28 -07:00
parent aa6448c30a
commit 41fab3c318
1 changed files with 6 additions and 5 deletions
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 import sys
from os import path
from contextlib import contextmanager from contextlib import contextmanager
from llama_cpp import Llama as LLM from llama_cpp import Llama as LLM
@@ -9,12 +10,12 @@ import ollama.prompt
@contextmanager @contextmanager
def suppress_stderr(): def suppress_stderr():
stderr = dup(sys.stderr.fileno()) stderr = os.dup(sys.stderr.fileno())
with open(devnull, "w") as devnull: with open(os.devnull, "w") as devnull:
dup2(devnull.fileno(), sys.stderr.fileno()) os.dup2(devnull.fileno(), sys.stderr.fileno())
yield yield
dup2(stderr, sys.stderr.fileno()) os.dup2(stderr, sys.stderr.fileno())
def generate(model, prompt, models_home=".", llms={}, *args, **kwargs): def generate(model, prompt, models_home=".", llms={}, *args, **kwargs):