Fixing the error facing in google colab
#6
by
Roy-Pentagon
- opened
README.md
CHANGED
@@ -44,7 +44,7 @@ device = "cuda"
|
|
44 |
|
45 |
tokenizer = AutoTokenizer.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base")
|
46 |
|
47 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base").to(
|
48 |
|
49 |
def paraphrase(
|
50 |
question,
|
|
|
44 |
|
45 |
tokenizer = AutoTokenizer.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base")
|
46 |
|
47 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("humarin/chatgpt_paraphraser_on_T5_base").to('cpu')
|
48 |
|
49 |
def paraphrase(
|
50 |
question,
|