xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix: fix flake8 check

This commit is contained in:
jiajingbin 2025-03-18 17:59:43 +08:00
parent 10aa645982
commit 181a4bb0fa
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tools/tdgpt/dockerfile/tdgpt/taos_ts_server.py
View File

@ -97,6 +97,9 @@ class TaosConfig(PretrainedConfig):
self.Taos_max_position_embeddings = Taos_max_position_embeddings self.Taos_max_position_embeddings = Taos_max_position_embeddings
super().__init__(**kwargs) super().__init__(**kwargs)
class BaseStreamer:
pass
class TaosTSGenerationMixin(GenerationMixin): class TaosTSGenerationMixin(GenerationMixin):
@torch.no_grad() @torch.no_grad()
def generate( def generate(
@ -730,6 +733,15 @@ def init_model():
#seqs = seqs.to(device) #seqs = seqs.to(device)
print(Taos_model) print(Taos_model)
def train():
pass
def infer():
pass
def data_view():
pass
@app.route('/get_train_data', methods=['POST']) @app.route('/get_train_data', methods=['POST'])
def get_train_data(): def get_train_data():
try: try: