Make the code more like PEP8 for readability (#862)

This commit is contained in:
oobabooga 2023-04-07 00:15:45 -03:00 committed by GitHub
parent 848c4edfd5
commit ea6e77df72
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
28 changed files with 302 additions and 165 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: