matt/ollama37
1
0
Fork 0
mirror of https://github.com/dogkeeper886/ollama37.git synced 2025-12-12 00:37:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

clean up pull

Browse Source
This commit is contained in:
Bruce MacDonald
2023-06-29 15:06:34 -04:00
parent d57903875e
commit 61e39bf5d9
3 changed files with 57 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ollama/prompt.py
View File

@@ -1,4 +1,4 @@
from os import path
import os
from difflib import SequenceMatcher
from jinja2 import Environment, PackageLoader
@@ -9,8 +9,8 @@ def template(model, prompt):
environment = Environment(loader=PackageLoader(__name__, 'templates'))
for template in environment.list_templates():
base, _ = path.splitext(template)
ratio = SequenceMatcher(None, model.lower(), base).ratio()
base, _ = os.path.splitext(template)
ratio = SequenceMatcher(None, os.path.basename(model.lower()), base).ratio()
if ratio > best_ratio:
best_ratio = ratio
best_template = template