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

View File

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