AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Release 8-bit models memory

Browse Source
This commit is contained in:
oobabooga 2023-01-19 12:01:58 -03:00
parent f9faad4cfa
commit 759da435e3
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server.py
View File

@ -11,6 +11,7 @@ import transformers
from html_generator import * from html_generator import *
from transformers import AutoTokenizer, AutoModelForCausalLM from transformers import AutoTokenizer, AutoModelForCausalLM
import warnings import warnings
import gc
transformers.logging.set_verbosity_error() transformers.logging.set_verbosity_error()
@ -151,6 +152,7 @@ def generate_reply(question, tokens, inference_settings, selected_model, eos_tok
model = None model = None
tokenizer = None tokenizer = None
if not args.cpu: if not args.cpu:
gc.collect()
torch.cuda.empty_cache() torch.cuda.empty_cache()
model, tokenizer = load_model(model_name) model, tokenizer = load_model(model_name)
if inference_settings != loaded_preset: if inference_settings != loaded_preset: