Update README.md

README.md

@@ -83,7 +83,7 @@ The following examples was obtained with `temperature = 0.85`.
 
 <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
 
-CelineGPT-12B-240831 was trained on a diverse, *French-only* corpus consisting of:
+CelineGPT-12B-v0.1 was trained on a diverse, *French-only* corpus consisting of:
 - All available interviews (TV, radio, written) given by Céline;
 - A significant portion of Céline's correspondence, manually transformed into question-and-answer pairs;
 - Portions of Céline's post-war works, notably the dialogues in 'Conversations with Professor Y,' 'Castle to Castle,' 'North,' 'Rigadoon,' 'Fable for Another Time,' and various other texts.
@@ -112,7 +112,7 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig, TextStreamer
 
 # The HF repo of the model
-model_id = "krogoldAI/CelineGPT-12B-240831"
+model_id = "krogoldAI/CelineGPT-12B-v0.1"
 
 # Use GPU when available
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")