Make the code more like PEP8

This commit is contained in:
oobabooga
2023-04-07 00:00:16 -03:00
parent 8e4e8dd741
commit 7b1b1d8e39
30 changed files with 314 additions and 177 deletions

View File

@@ -1,13 +1,11 @@
'''
Based on
Based on
https://github.com/abetlen/llama-cpp-python
Documentation:
https://abetlen.github.io/llama-cpp-python/
'''
import multiprocessing
from llama_cpp import Llama
from modules import shared
@@ -31,7 +29,7 @@ class LlamaCppModel:
self.model = Llama(**params)
# This is ugly, but the model and the tokenizer are the same object in this library.
return result, result
return result, result
def encode(self, string):
if type(string) is str: