Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 08:07:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

Change openai models list result output to match the dummy model example.

Browse Source 
Change openai models list result output to match the dummy model example.
This commit is contained in:
Jonathan Sowards 2024-03-17 19:38:05 -06:00 committed by GitHub
parent 0a67325809
commit e0d466ed14
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
extensions/openai/models.py
View File

@ -14,7 +14,15 @@ def get_current_model_info():
def list_models():
return {'model_names': get_available_models()[1:]}
result = {
"object": "list",
"data": []
}
# get models and add them to the result
for model in get_available_models():
result["data"].append(model_info_dict(model))
return {'model_names': result}
def list_dummy_models():