Update README.md

README.md

@@ -11,10 +11,10 @@ tags:
 ---
 # Model Card for Model ID
 
-This is the adapters from the LoRa fine-tuning of the phi-2 model from Microsoft. It was trained on the Electrical-engineering dataset.
+This is the adapters from the LoRa fine-tuning of the phi-2 model from Microsoft. It was trained on the Electrical-engineering dataset combined with garage-bAInd/Open-Platypus.
 
 - **Developed by:** STEM.AI
-- **Model type:** Q&A
+- **Model type:** Q&A and code generation
 - **Language(s) (NLP):** English
 - **Finetuned from model [optional]:** microsoft/phi-2
 
@@ -28,16 +28,19 @@ Q&A related to electrical engineering, and Kicad software. Creation of Python co
 
 ### Training Data
 
-<!-- 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. -->
+Dataset related to electrical engineering: STEM-AI-mtl/Electrical-engineering
 
-[More Information Needed]
+Combined with
+
+Dataset related to STEM and NLP: garage-bAInd/Open-Platypus
 
 ### Training Procedure 
 
-A LoRa PEFT was performed on a 48 Gb Nvidia GPU.
+A LoRa PEFT was performed on a 48 Gb A40 Nvidia GPU.
 
 ## Model Card Authors [optional]
 
 STEM.AI: stem.ai.mtl@gmail.com
+William Harbec