Update README.md

README.md

@@ -57,7 +57,7 @@ language:
 - en
 ---
 
-# Phi-3-mini-128k-instruct
+# Rubra Phi-3 Mini 128k Instruct
 
 ## Model description
 The model is the result of further post-training [microsoft/Phi-3-mini-128k-instruct](https://huggingface.co/microsoft/Phi-3-mini-128k-instruct). This model is designed for high performance in various instruction-following tasks and complex interactions, including multi-turn function calling and detailed conversations.
@@ -90,7 +90,7 @@ model_id = "rubra-ai/Phi-3-mini-128k-instruct"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
-    torch_dtype=torch.bfloat16,
+    torch_dtype="auto",
     device_map="auto",
 )
 ```