BAAI
/

AquilaChat-7B

Inference Endpoints

Model card Files Files and versions Community

shunxing1234 commited on Jul 3, 2023

Commit

31c9df4

•

1 Parent(s): c61e7d4

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -115,7 +115,7 @@ processors.append(TopPLogitsWarper(0.95))
 # Copied from https://github.com/bojone/NBCE/blob/main/test.py#L51-L106
 @torch.inference_mode()
 def generate(max_tokens, batch):
-    """Naive Bayes-based Context Extension 演示代码
     """
     inputs = tokenizer(batch, padding='longest', return_tensors='pt').to(device)
     input_ids = inputs.input_ids

 # Copied from https://github.com/bojone/NBCE/blob/main/test.py#L51-L106
 @torch.inference_mode()
 def generate(max_tokens, batch):
+    """Naive Bayes-based Context Extension
     """
     inputs = tokenizer(batch, padding='longest', return_tensors='pt').to(device)
     input_ids = inputs.input_ids