nicholasKluge
/

Aira-2-774M

Text Generation

instruction tuned

text generation

Carbon Emissions

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

nicholasKluge commited on Jun 10, 2023

Commit

a2398f6

•

1 Parent(s): 402c200

Upload AIRA-Fine-Tuning-gpt2.ipynb

Files changed (1) hide show

AIRA-Fine-Tuning-gpt2.ipynb +1 -1

AIRA-Fine-Tuning-gpt2.ipynb CHANGED Viewed

@@ -1124,8 +1124,8 @@
         "tokenizer = AutoTokenizer.from_pretrained('nicholasKluge/Aira-Instruct-124M')\n",
         "aira = AutoModelForCausalLM.from_pretrained('nicholasKluge/Aira-Instruct-124M') \n",
         "\n",
-        "aira.to(device)\n",
         "aira.eval()\n",
         "\n",
         "question = input(\"Enter your question: \")\n",
         "\n",

         "tokenizer = AutoTokenizer.from_pretrained('nicholasKluge/Aira-Instruct-124M')\n",
         "aira = AutoModelForCausalLM.from_pretrained('nicholasKluge/Aira-Instruct-124M') \n",
         "\n",
         "aira.eval()\n",
+        "aira.to(device)\n",
         "\n",
         "question = input(\"Enter your question: \")\n",
         "\n",