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
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):
......
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