Merge pull request #305 from SillyLossy/united

Fix setting softprompt via V1 API
This commit is contained in:
henk717
2023-03-09 13:03:14 +01:00
committed by GitHub

View File

@@ -10510,7 +10510,7 @@ def soft_prompt_validator(soft_prompt: str):
raise ValidationError("Cannot use soft prompts with current backend.") raise ValidationError("Cannot use soft prompts with current backend.")
if any(q in soft_prompt for q in ("/", "\\")): if any(q in soft_prompt for q in ("/", "\\")):
return return
z, _, _, _, _ = fileops.checksp(soft_prompt.strip(), koboldai_vars.modeldim) z, _, _, _, _ = fileops.checksp("./softprompts/"+soft_prompt.strip(), koboldai_vars.modeldim)
if isinstance(z, int): if isinstance(z, int):
raise ValidationError("Must be a valid soft prompt name.") raise ValidationError("Must be a valid soft prompt name.")
z.close() z.close()