diff --git a/src/cogs/ollama.py b/src/cogs/ollama.py index 0118c1f..2f6e770 100644 --- a/src/cogs/ollama.py +++ b/src/cogs/ollama.py @@ -533,7 +533,11 @@ class Ollama(commands.Cog): self.log.debug("Resolved model to %r" % model) 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( f"{ctx.user.mention}: You can only use images with llava. Switching model to `llava:latest`.", delete_after=5