Update README.md

README.md

@@ -25,7 +25,7 @@ tags:
 - **Model type:** Causal decoder-only
 - **Architecture:** Transformer-base
 - **Language(s) (NLP):** Mainly English
-- **License:** TII Falcon-Mamba License 2.0
+- **License:** TII Falcon-LLM License 2.0
 
 <br>
 
@@ -64,8 +64,8 @@ print(tokenizer.decode(outputs[0]))
 # pip install accelerate
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("tiiuae/Falcon3-7B-Base")
-model = AutoModelForCausalLM.from_pretrained("tiiuae/Falcon3-7B-Base", device_map="auto")
+tokenizer = AutoTokenizer.from_pretrained("tiiuae/Falcon3-10B-Base")
+model = AutoModelForCausalLM.from_pretrained("tiiuae/Falcon3-10B-Base", device_map="auto")
 
 input_text = "Question: How many hours in one day? Answer: "
 input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")
@@ -85,8 +85,8 @@ print(tokenizer.decode(outputs[0]))
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("tiiuae/Falcon3-7B-Base")
-model = AutoModelForCausalLM.from_pretrained("tiiuae/Falcon3-7B-Base", torch_dtype=torch.bfloat16).to(0)
+tokenizer = AutoTokenizer.from_pretrained("tiiuae/Falcon3-10B-Base")
+model = AutoModelForCausalLM.from_pretrained("tiiuae/Falcon3-10B-Base", torch_dtype=torch.bfloat16).to(0)
 
 model = torch.compile(model)