This commit is contained in:
Michael Yang
2023-06-27 12:19:34 -07:00
parent 8750e9889f
commit 1ae91f75f0
7 changed files with 196 additions and 216 deletions

9
ollama/model.py Normal file
View File

@@ -0,0 +1,9 @@
from os import walk, path
def models(models_home='.', *args, **kwargs):
for root, _, files in walk(models_home):
for file in files:
base, ext = path.splitext(file)
if ext == '.bin':
yield base, path.join(root, file)