Skip to content

Commit

Permalink
fix: parse url escaped values for model names
Browse files Browse the repository at this point in the history
This will allow the `/v2/models/{model_name}` endpoint to support forward slashes in the model name, as is common for text models. For example, `aphrodite%2FNeverSleep%2FNoromaid-13b-v0.3` will become `aphrodite/NeverSleep/Noromaid-13b-v0.3`.
  • Loading branch information
tazlin committed Mar 15, 2024
1 parent a0ac730 commit a913ea8
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions horde/database/functions.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
import json
import time
import urllib.parse
import uuid
from datetime import datetime, timedelta

Expand Down Expand Up @@ -256,6 +257,12 @@ def worker_exists(worker_id):
def get_available_models(filter_model_name: str = None):
models_dict = {}
available_worker_models = None

if filter_model_name is not None:
# Decode the filter_model_name from URL encoding
# e.g., `aphrodite%2FNeverSleep%2FNoromaid-13b-v0.3` will become `aphrodite/NeverSleep/Noromaid-13b-v0.3`.
filter_model_name = urllib.parse.unquote(filter_model_name)

for model_type, worker_class, wp_class, procgen_class in [
("image", ImageWorker, ImageWaitingPrompt, ImageProcessingGeneration),
("text", TextWorker, TextWaitingPrompt, TextProcessingGeneration),
Expand Down

0 comments on commit a913ea8

Please sign in to comment.