nex/college-bot-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Activity Actions

Allow other llava types
All checks were successful
Build and Publish Jimmy.2 / build_and_publish (push) Successful in 6s
Details

Browse source
This commit is contained in:
Nexus 2024-05-27 18:30:51 +01:00
parent 22789057f8
commit 4dc0e6418f
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/cogs/ollama.py
View file

@ -533,7 +533,11 @@ class Ollama(commands.Cog):
self.log.debug("Resolved model to %r" % model) self.log.debug("Resolved model to %r" % model)
if image: if image:
if fnmatch(model, "llava:*") is False: patterns = [
"llava:*",
"llava-llama*:*"
]
if any(fnmatch(model, p) for p in patterns) is False:
await ctx.send( await ctx.send(
f"{ctx.user.mention}: You can only use images with llava. Switching model to `llava:latest`.", f"{ctx.user.mention}: You can only use images with llava. Switching model to `llava:latest`.",
delete_after=5 delete_after=5