Update README.md

README.md

@@ -14,14 +14,14 @@ license: apache-2.0
 
 Nayana_base_combined_v1
 
-```
+```python
 from transformers import AutoModel, AutoTokenizer
 from peft import PeftModel, PeftConfig, AutoPeftModelForCausalLM
 from transformers import AutoModelForCausalLM
 import torch
 
-tokenizer = AutoTokenizer.from_pretrained('v1v1d/Nayana_base_combined', trust_remote_code=True , torch_dtype=torch.float16)
-model = AutoModel.from_pretrained('v1v1d/Nayana_base_combined', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True, pad_token_id=tokenizer.eos_token_id , torch_dtype=torch.float16)
+tokenizer = AutoTokenizer.from_pretrained('v1v1d/Nayana_base_combined_lora_64', trust_remote_code=True , torch_dtype=torch.float16)
+model = AutoModel.from_pretrained('v1v1d/Nayana_base_combined_lora_64', trust_remote_code=True, low_cpu_mem_usage=True, device_map='cuda', use_safetensors=True, pad_token_id=tokenizer.eos_token_id , torch_dtype=torch.float16)
 
 
 model = model.eval().cuda()