Update README.md

README.md

@@ -86,8 +86,8 @@ export HF_HUB_ENABLE_HF_TRANSFER=1
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-tokenizer = AutoTokenizer.from_pretrained("databricks/dbrx-base", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("databricks/dbrx-base", device_map="cpu", torch_dtype=torch.bfloat16, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("Undi95/dbrx-base", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("Undi95/dbrx-base", device_map="cpu", torch_dtype=torch.bfloat16, trust_remote_code=True)
 
 input_text = "Databricks was founded in "
 input_ids = tokenizer(input_text, return_tensors="pt")
@@ -101,8 +101,8 @@ print(tokenizer.decode(outputs[0]))
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-tokenizer = AutoTokenizer.from_pretrained("databricks/dbrx-base", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("databricks/dbrx-base", device_map="auto", torch_dtype=torch.bfloat16, trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("Undi95/dbrx-base", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("Undi95/dbrx-base", device_map="auto", torch_dtype=torch.bfloat16, trust_remote_code=True)
 
 input_text = "Databricks was founded in "
 input_ids = tokenizer(input_text, return_tensors="pt").to("cuda")