Update README.md

README.md

@@ -87,7 +87,11 @@ model = AutoModelForCausalLM.from_pretrained(
     torch_dtype=torch.float16,
     device_map="auto"
 )
-model = PeftModel.from_pretrained(model, MODEL_NAME)
+model = PeftModel.from_pretrained(
+    model,
+    MODEL_NAME,
+    torch_dtype=torch.float16
+)
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 generation_config = GenerationConfig.from_pretrained(MODEL_NAME)
 print(generation_config)