Update README.md

README.md

@@ -11,7 +11,6 @@ pipeline_tag: text-generation
 library_name: transformers
 ---
 
-
 ## 💻 For English
 Megatron_llama3_2x8B is a Mixure of Experts (MoE) (two llama3 models)
 
@@ -28,8 +27,8 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto", load_in_8bit= True)
 
 messages = [
-    {"role": "system", "content": "You are a helpful chatbot who always responds friendly."},
-    {"role": "user", "content": "f(x)=3x^2+4x+12 so what is f(3)?"},
+    {"role": "system", "content": "You are a helpful chatbot, named Defne, who always responds friendly."},
+    {"role": "user", "content": "Answer the questions: 1) Who are you? 2) f(x)=3x^2+4x+12 so what is f(3)?"},
 ]
 
 input_ids = tokenizer.apply_chat_template(messages, return_tensors="pt").to("cuda")
@@ -45,12 +44,24 @@ outputs = model.generate(input_ids,
 response = outputs[0][input_ids.shape[-1]:]
 print(tokenizer.decode(response, skip_special_tokens=True))
 ```
+### Output
+```
+Hello there! I'm Defne, a friendly chatbot here to help with any questions you may have.
 
-# Megatron_llama3_2x8B
+Now, let's get to the math problem!
 
+The function is f(x) = 3x^2 + 4x + 12, and we want to find f(3). To do that, we can plug in 3 for x in the function:
 
+f(3) = 3(3)^2 + 4(3) + 12
+f(3) = 3(9) + 12 + 12
+f(3) = 27 + 24
+f(3) = 51
+
+So, f(3) is equal to 51!
+```
 
 ## 💻 Türkçe İçin
+Megatron_llama3_2x8B, iki llama3 8B modelinin birleşmesi ile oluşturulan MoE yapısında bir modeldir.  
 
 ```python
 !pip install -qU transformers bitsandbytes accelerate
@@ -61,7 +72,7 @@ import torch
 model_id = "Eurdem/Megatron_llama3_2x8B"
 
 tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto", load_in_4bit= True)
+model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto", load_in_8bit= True)
 
 messages = [
     {"role": "system", "content": "Sen Defne isimli Türkçe konuşan bir chatbotsun."},
@@ -81,9 +92,9 @@ outputs = model.generate(input_ids,
 response = outputs[0][input_ids.shape[-1]:]
 print(tokenizer.decode(response, skip_special_tokens=True))
 ```
-
 ### Çıktı
-```Merhaba! Ben Sen Defne, Türkçe konuşan bir chatbotum. Hizmetinizdeyim.
+```
+Merhaba! Ben Sen Defne, Türkçe konuşan bir chatbotum. Hizmetinizdeyim.
 
 Sorunuzun 2. kısmı için, f(x) = 3x^2 + 4x + 12 formülünü ele alalım. f(3)'ün hesabını yapalım:
 
@@ -92,4 +103,5 @@ f(3) = 3(3)^2 + 4(3) + 12
 = 27 + 24
 = 51
 
-Bu nedenle, f(3) 51'dir.```
+Bu nedenle, f(3) 51'dir.
+```