About "seqio" in "hf_model.py"
censhallwe opened this issue · 0 comments
Describe the bug
Traceback (most recent call last):
File "/nvme/liyuanchun/wangweijun/T5/text-to-text-transfer-transformer/hf_model_test.py", line 33, in
model.predict(
File "/nvme/liyuanchun/wangweijun/T5/text-to-text-transfer-transformer/t5/models/hf_model.py", line 505, in predict
vocab = seqio.get_default_vocabulary()
AttributeError: module 'seqio' has no attribute 'get_default_vocabulary'
I use "seqio"==0.0.15
In my "hf_model_test.py", I tried to predict, however, I met this error and couldn't find solution.
The code is as follow:
inputs = [
"translate English to German: how many states does the US has? ",
"translate English to German: who is the US president?",
"translate English to German: who got the first nobel prize in physics?",
"translate English to German: when is the next deadpool movie being released?",
"translate English to German: which mode is used for short wave broadcast service?",
"translate English to German: the south west wind blows across nigeria between?",
]
model.predict(
inputs,
sequence_length={"inputs": 32},
batch_size=2,
output_file=f"{path}/example_predictions.txt",
)