Update README.md

README.md

@@ -99,17 +99,17 @@ print(tokenizer.decode(outputs[0]))
 
 </details>
 
-#### INT8
+#### 4-bit
 
 <details>
 <summary> Click to expand </summary>
 
 ```python
 # pip install bitsandbytes accelerate
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 
 tokenizer = AutoTokenizer.from_pretrained("tiiuae/sindibad-7b")
-model = AutoModelForCausalLM.from_pretrained("tiiuae/sindibad-7b", device_map="auto", load_in_8bit=True)
+model = AutoModelForCausalLM.from_pretrained("tiiuae/sindibad-7b", device_map="auto", quantization_config=BitsAndBytesConfig(load_in_4bit=True))
 
 input_text = "Question: How many hours in one day? Answer: "
 input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
@@ -123,18 +123,20 @@ print(tokenizer.decode(outputs[0]))
 
 # Training Details
 
-## Training Data
+Jingwei
 
+## Training Data
 
+Guillaume
 
 ## Training Procedure
 
+Maksim
 
 # Evaluation
 
-## Testing Data, Factors & Metrics
-
+Dhia & Youne
 
 ## Results 
 
-
+Ilyas