"docs/source/vscode:/vscode.git/clone" did not exist on "413b71032e20df1a1a99b0a63ee7b227a89b5683"
Commit 41fab3c3 authored by Michael Yang's avatar Michael Yang
Browse files

fix imports

parent aa6448c3
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