Merge remote-tracking branch 'refs/remotes/origin/main'

This commit is contained in:
oobabooga 2023-09-17 18:01:27 -07:00
commit 280cca9f66
5 changed files with 9 additions and 7 deletions

View File

@ -101,9 +101,7 @@ pip install -r requirements_nocuda.txt
``` ```
cd text-generation-webui cd text-generation-webui
mkdir repositories git clone https://github.com/turboderp/exllama repositories/exllama
cd repositories
git clone https://github.com/turboderp/exllama
``` ```
#### bitsandbytes on older NVIDIA GPUs #### bitsandbytes on older NVIDIA GPUs

View File

@ -14,7 +14,7 @@ try:
from exllama.model import ExLlama, ExLlamaCache, ExLlamaConfig from exllama.model import ExLlama, ExLlamaCache, ExLlamaConfig
from exllama.tokenizer import ExLlamaTokenizer from exllama.tokenizer import ExLlamaTokenizer
except: except:
logger.warning('Exllama module failed to load. Will attempt to load from repositories.') logger.warning('exllama module failed to import. Will attempt to import from repositories/.')
try: try:
from modules.relative_imports import RelativeImport from modules.relative_imports import RelativeImport
@ -23,7 +23,10 @@ except:
from model import ExLlama, ExLlamaCache, ExLlamaConfig from model import ExLlama, ExLlamaCache, ExLlamaConfig
from tokenizer import ExLlamaTokenizer from tokenizer import ExLlamaTokenizer
except: except:
logger.error("Could not find repositories/exllama/. Make sure that exllama is cloned inside repositories/ and is up to date.") logger.error(
"Could not find repositories/exllama. Please ensure that exllama"
" (https://github.com/turboderp/exllama) is cloned inside repositories/ and is up to date."
)
raise raise

View File

@ -27,6 +27,7 @@ def apply_settings(extension, name):
def load_extensions(): def load_extensions():
global state, setup_called global state, setup_called
state = {}
for i, name in enumerate(shared.args.extensions): for i, name in enumerate(shared.args.extensions):
if name in available_extensions: if name in available_extensions:
if name != 'api': if name != 'api':

View File

@ -8,7 +8,7 @@ accelerate==0.22.*
colorama colorama
datasets datasets
einops einops
exllamav2==0.0.1 exllamav2==0.0.2
markdown markdown
numpy==1.24 numpy==1.24
optimum==1.13.1 optimum==1.13.1

View File

@ -8,7 +8,7 @@ accelerate==0.22.*
colorama colorama
datasets datasets
einops einops
exllamav2==0.0.1 exllamav2==0.0.2
markdown markdown
numpy==1.24 numpy==1.24
optimum==1.13.1 optimum==1.13.1