Update Codes/02 - Run the model in Jypiter.ipynb
Browse files
Codes/02 - Run the model in Jypiter.ipynb
CHANGED
@@ -53,7 +53,7 @@
|
|
53 |
"from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig, TextStreamer\n",
|
54 |
"\n",
|
55 |
"# The HF repo of the model\n",
|
56 |
-
"model_id = \"krogoldAI/CelineGPT-12B-
|
57 |
]
|
58 |
},
|
59 |
{
|
|
|
53 |
"from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig, TextStreamer\n",
|
54 |
"\n",
|
55 |
"# The HF repo of the model\n",
|
56 |
+
"model_id = \"krogoldAI/CelineGPT-12B-v0.1\""
|
57 |
]
|
58 |
},
|
59 |
{
|