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

Add /v1/internal/model-info endpoint

Browse source
This commit is contained in:
oobabooga 2023-11-07 18:59:02 -08:00
parent 6e2e0317af
commit f6ca9cfcdc
3 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
extensions/openai/models.py
View file

@ -7,6 +7,13 @@ from modules.models_settings import get_model_metadata, update_model_parameters
from modules.utils import get_available_models
def get_current_model_info():
return {
'model_name': shared.model_name,
'lora_names': shared.lora_names
}
def get_current_model_list() -> list:
return [shared.model_name] # The real chat/completions model, maybe "None"