Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -9,6 +9,8 @@ Note: this is a base model and not meant to be used as is. We recommend first fi
9
 
10
  ```
11
  # Usage
 
 
12
  tokenizer = AutoTokenizer.from_pretrained('sarvamai/OpenHathi-7B-Hi-v0.1-Base')
13
  model = LlamaForCausalLM.from_pretrained('sarvamai/OpenHathi-7B-Hi-v0.1-Base', torch_dtype=torch.bfloat16)
14
 
 
9
 
10
  ```
11
  # Usage
12
+ from transformers import AutoTokenizer,LlamaForCausalLM
13
+ import torch
14
  tokenizer = AutoTokenizer.from_pretrained('sarvamai/OpenHathi-7B-Hi-v0.1-Base')
15
  model = LlamaForCausalLM.from_pretrained('sarvamai/OpenHathi-7B-Hi-v0.1-Base', torch_dtype=torch.bfloat16)
16