sanjay920 commited on
Commit
62dbd0b
1 Parent(s): 125fd6d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -57,7 +57,7 @@ language:
57
  - en
58
  ---
59
 
60
- # Phi-3-mini-128k-instruct
61
 
62
  ## Model description
63
  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"
90
  tokenizer = AutoTokenizer.from_pretrained(model_id)
91
  model = AutoModelForCausalLM.from_pretrained(
92
  model_id,
93
- torch_dtype=torch.bfloat16,
94
  device_map="auto",
95
  )
96
  ```
 
57
  - en
58
  ---
59
 
60
+ # Rubra Phi-3 Mini 128k Instruct
61
 
62
  ## Model description
63
  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
  tokenizer = AutoTokenizer.from_pretrained(model_id)
91
  model = AutoModelForCausalLM.from_pretrained(
92
  model_id,
93
+ torch_dtype="auto",
94
  device_map="auto",
95
  )
96
  ```