model: add model file

This commit is contained in:
csunny 2023-04-28 22:04:37 +08:00
parent 38f57e157c
commit c72ae1a87f
3 changed files with 8 additions and 0 deletions

1
.gitignore vendored
View File

@ -20,6 +20,7 @@ parts/
sdist/
var/
wheels/
models/
pip-wheel-metadata/
share/python-wheels/
*.egg-info/

View File

@ -37,3 +37,4 @@ class ModerLoader:
return model, tokenizer

View File

@ -1,3 +1,9 @@
#!/usr/bin/env python3
#-*- coding: utf-8 -*-
from model.loader import ModerLoader
from fastchat.serve.inference import generate_stream
from configs.model_config import *
if __name__ == "__main__":
pass