Update README.md (#6)

- Update README.md (22eb92898d63ebd3fb3f5a4764aa0e7798f21b41)


Co-authored-by: wbag <[email protected]>

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")