Skip to content

GitLab

Unverified Commit 42047b15 authored by Michael Yang's avatar Michael Yang Committed by GitHub
Browse files

Merge pull request #23 from jmorganca/fix-imports 

fix imports
parents aa6448c3 41fab3c3
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 5 deletions
ollama/engine.py
View file @ 42047b15
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 ...@@ -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):
......
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