baalajimaestro/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix: fileno error google colab (#729) (#1156)

Browse source 
Instead of using a devnull just made a dummy class with a 'write()' method that does nothing.
This commit is contained in:
Dulsara 2024-02-02 22:35:46 +05:30 committed by GitHub
parent 8a5911bd5d
commit bebfba0f08
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
llama_cpp/_utils.py
View file

@ -4,9 +4,9 @@ import sys
import sys import sys
from typing import Any, Dict from typing import Any, Dict
# Avoid "LookupError: unknown encoding: ascii" when open() called in a destructor class NullDevice():
outnull_file = open(os.devnull, "w") def write(self, s):
errnull_file = open(os.devnull, "w") pass
class suppress_stdout_stderr(object): class suppress_stdout_stderr(object):
# NOTE: these must be "saved" here to avoid exceptions when using # NOTE: these must be "saved" here to avoid exceptions when using
@ -21,42 +21,20 @@ class suppress_stdout_stderr(object):
def __enter__(self): def __enter__(self):
if self.disable: if self.disable:
return self return self
# Check if sys.stdout and sys.stderr have fileno method
if not hasattr(self.sys.stdout, 'fileno') or not hasattr(self.sys.stderr, 'fileno'):
return self # Return the instance without making changes
self.old_stdout_fileno_undup = self.sys.stdout.fileno()
self.old_stderr_fileno_undup = self.sys.stderr.fileno()
self.old_stdout_fileno = self.os.dup(self.old_stdout_fileno_undup)
self.old_stderr_fileno = self.os.dup(self.old_stderr_fileno_undup)
self.old_stdout = self.sys.stdout self.old_stdout = self.sys.stdout
self.old_stderr = self.sys.stderr self.old_stderr = self.sys.stderr
self.os.dup2(outnull_file.fileno(), self.old_stdout_fileno_undup) self.sys.stdout = NullDevice()
self.os.dup2(errnull_file.fileno(), self.old_stderr_fileno_undup) self.sys.stderr = NullDevice()
self.sys.stdout = outnull_file
self.sys.stderr = errnull_file
return self return self
def __exit__(self, *_): def __exit__(self, *_):
if self.disable: if self.disable:
return return
# Check if sys.stdout and sys.stderr have fileno method
if hasattr(self.sys.stdout, 'fileno') and hasattr(self.sys.stderr, 'fileno'):
self.sys.stdout = self.old_stdout self.sys.stdout = self.old_stdout
self.sys.stderr = self.old_stderr self.sys.stderr = self.old_stderr
self.os.dup2(self.old_stdout_fileno, self.old_stdout_fileno_undup)
self.os.dup2(self.old_stderr_fileno, self.old_stderr_fileno_undup)
self.os.close(self.old_stdout_fileno)
self.os.close(self.old_stderr_fileno)
class MetaSingleton(type): class MetaSingleton(type):
""" """