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

Automatically switch model to llava when an image is attached

Browse source
This commit is contained in:
Nexus 2024-01-11 13:21:41 +00:00
parent 789231f2da
commit ae065c8d97
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/cogs/ollama.py
View file

@ -146,9 +146,13 @@ class Ollama(commands.Cog):
if image:
if fnmatch(model, "llava:*") is False:
await ctx.respond("You can only use images with llava.")
return
elif image.size > 1024 * 1024 * 25:
await ctx.respond(
"You can only use images with llava. Switching model to `llava:latest`.",
delete_after=5
)
model = "llava:latest"
if image.size > 1024 * 1024 * 25:
await ctx.respond("Attachment is too large. Maximum size is 25 MB, for sanity. Try compressing it.")
return
elif not fnmatch(image.content_type, "image/*"):