Update README.md

README.md

@@ -42,7 +42,9 @@ The training process was monitored using `wandb`, with a brief interruption wher
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-device = 'cuda:0'
+device = "cuda:0"
+model_name = "Nagi-ovo/lama-3-8b-sft-ruozhiba"
+
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
     load_in_4bit=True, bnb_4bit_quant_type="nf4",