StarRing2022
/

RWKV-4-World-1.5B

Inference Endpoints

Model card Files Files and versions Community

StarRing2022 commited on Jun 26, 2023

Commit

28ac2d1

·

1 Parent(s): ad471b3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ out = model.generate(input_ids,max_new_tokens=40)<br><br>
 outlist = out[0].tolist()<br>
 for i  in outlist:<br>
-  &nbsp;&nbsp;&nbsp;&nbsp;if i==0:<br> #要删除tokenid为0的元素
   &nbsp;&nbsp;&nbsp;&nbsp;outlist.remove(i)<br>
 answer = tokenizer.decode(outlist)<br>

 outlist = out[0].tolist()<br>
 for i  in outlist:<br>
+  &nbsp;&nbsp;&nbsp;&nbsp;if i==0:&nbsp;#要删除tokenid为0的元素 <br>
   &nbsp;&nbsp;&nbsp;&nbsp;outlist.remove(i)<br>
 answer = tokenizer.decode(outlist)<br>