Fixing the error facing in google colab

#6
by Roy-Pentagon - opened
Files changed (1) hide show
  1. README.md +1 -1
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(device)
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,