davidberenstein1957 HF staff commited on
Commit
02de6f4
·
1 Parent(s): a8cfebd

update example

Browse files
examples/fine-tune-smollm2-on-synthetic-data.ipynb CHANGED
@@ -137,7 +137,7 @@
137
  ],
138
  "source": [
139
  "from transformers import pipeline\n",
140
- "# Let's test the base model before training\n",
141
  "prompt = \"What is the primary function of mitochondria within a cell?\"\n",
142
  "\n",
143
  "pipe = pipeline(\"text-generation\", model=model, tokenizer=tokenizer, device=device)\n",
@@ -261,6 +261,7 @@
261
  ],
262
  "source": [
263
  "prompt = \"What is the primary function of mitochondria within a cell?\"\n",
 
264
  "generator = pipeline(\n",
265
  " \"text-generation\",\n",
266
  " model=\"argilla/SmolLM2-360M-synthetic-concise-reasoning\",\n",
 
137
  ],
138
  "source": [
139
  "from transformers import pipeline\n",
140
+ "\n",
141
  "prompt = \"What is the primary function of mitochondria within a cell?\"\n",
142
  "\n",
143
  "pipe = pipeline(\"text-generation\", model=model, tokenizer=tokenizer, device=device)\n",
 
261
  ],
262
  "source": [
263
  "prompt = \"What is the primary function of mitochondria within a cell?\"\n",
264
+ "\n",
265
  "generator = pipeline(\n",
266
  " \"text-generation\",\n",
267
  " model=\"argilla/SmolLM2-360M-synthetic-concise-reasoning\",\n",