Update README.md

README.md

@@ -101,9 +101,9 @@ def inference(input_prompts, model, tokenizer):
 
 model_name = "ai4bharat/Airavata"
 
-tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side="left", token=hf_token)
+tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side="left")
 tokenizer.pad_token = tokenizer.eos_token
-model = AutoModelForCausalLM.from_pretrained(model_name, token=hf_token).to(device)
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16).to(device)
 
 input_prompts = [
     "मैं अपने समय प्रबंधन कौशल को कैसे सुधार सकता हूँ? मुझे पांच बिंदु बताएं।",