SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

refactored code a bit and made more generic

Browse Source
This commit is contained in:
Patrick von Platen
2020-03-05 14:56:56 +01:00
parent d8e2b3c547
commit c0d9dd3ba9
4 changed files with 28 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_bart.py
View File

@@ -442,7 +442,7 @@ class BartModelIntegrationTest(unittest.TestCase):
tokens = tok.encode(text, return_tensors="pt").to(torch_device)
extra_len = 20
gen_tokens_1 = hf.generate_1(tokens, num_beams=4, max_length=extra_len,) # repetition_penalty=10.,
gen_tokens = hf.generate(tokens, num_beams=4, max_length=extra_len, do_sample=False) # repetition_penalty=10.,
gen_tokens = hf.generate(tokens, num_beams=4, max_length=extra_len + 2, do_sample=False) # repetition_penalty=10.,
print("1: {}".format(gen_tokens_1))
print("2: {}".format(gen_tokens))
ipdb.set_trace()