mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add more info messages for truncation / instruction template

Browse source
This commit is contained in:
oobabooga 2023-11-15 16:13:36 -08:00
parent 883701bc40
commit a85ce5f055
3 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
extensions/openai/models.py
View file

@ -1,4 +1,5 @@
from modules import shared
from modules.logging_colors import logger
from modules.models import load_model, unload_model
from modules.models_settings import get_model_metadata, update_model_parameters
from modules.utils import get_available_models
@ -62,3 +63,7 @@ def _load_model(data):
for k in settings:
if k in shared.settings:
shared.settings[k] = settings[k]
if k == 'truncation_length':
logger.info(f"TRUNCATION LENGTH (UPDATED): {shared.settings['truncation_length']}")
elif k == 'instruction_template':
logger.info(f"INSTRUCTION TEMPLATE (UPDATED): {shared.settings['instruction_template']}")