oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-04-04 22:27:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove unnecessary shared.previous_model_name variable

Browse source
This commit is contained in:
oobabooga 2024-07-28 18:35:11 -07:00
parent addcb52c56
commit 9dcff21da9
4 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
modules/logits.py
View file

@ -13,8 +13,8 @@ global_scores = None
def get_next_logits(*args, **kwargs):
if shared.args.idle_timeout > 0 and shared.model is None and shared.previous_model_name not in [None, 'None']:
shared.model, shared.tokenizer = load_model(shared.previous_model_name)
if shared.args.idle_timeout > 0 and shared.model is None and shared.model_name not in [None, 'None']:
shared.model, shared.tokenizer = load_model(shared.model_name)
needs_lock = not args[2] # use_samplers
if needs_lock: