mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-24 12:45:45 +00:00
model: add model file
This commit is contained in:
parent
38f57e157c
commit
c72ae1a87f
1
.gitignore
vendored
1
.gitignore
vendored
@ -20,6 +20,7 @@ parts/
|
||||
sdist/
|
||||
var/
|
||||
wheels/
|
||||
models/
|
||||
pip-wheel-metadata/
|
||||
share/python-wheels/
|
||||
*.egg-info/
|
||||
|
@ -37,3 +37,4 @@ class ModerLoader:
|
||||
|
||||
return model, tokenizer
|
||||
|
||||
|
||||
|
@ -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
|
Loading…
Reference in New Issue
Block a user