efederici
/

bart-lfqa-it

text2text-generation

Model card Files Files and versions

efederici commited on Feb 14, 2023

Commit

8764545

·

1 Parent(s): 8adbb6e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -10,7 +10,7 @@ This model is a fine-tuned version of [bart-it](https://huggingface.co/morenolq/
 ```python
 import torch
-from transformers import AutoTokenizer, AutoModel, AutoModelForSeq2SeqLM
 model_name = "efederici/bart-lfqa-it"
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

 ```python
 import torch
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 model_name = "efederici/bart-lfqa-it"
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')