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 image:
if fnmatch(model, "llava:*") is False: if fnmatch(model, "llava:*") is False:
await ctx.respond("You can only use images with llava.") await ctx.respond(
return "You can only use images with llava. Switching model to `llava:latest`.",
elif image.size > 1024 * 1024 * 25: 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.") await ctx.respond("Attachment is too large. Maximum size is 25 MB, for sanity. Try compressing it.")
return return
elif not fnmatch(image.content_type, "image/*"): elif not fnmatch(image.content_type, "image/*"):