Update README.md

README.md

@@ -12,14 +12,13 @@ license: mit
 ## How to Get Started with the Model
 ```python
   import torch
-  from transformers import AutoModelForCausalLM
+  from transformers import AutoModelForCausalLM, BitsAndBytesConfig
   from peft import PeftModel, PeftConfig
-  from transformers import AutoModelForCausalLM
 
   config = PeftConfig.from_pretrained("yamete4/codegen-350M-mono-QLoRa-flytech")
   model = AutoModelForCausalLM.from_pretrained("shpotes/codegen-350M-mono",
-                                                quantization_config=config,)
-  model = PeftModel.from_pretrained(model, "yamete4/codegen-350M-mono-QLoRa-flytech")
+                                                quantization_config=BitsAndBytesConfig(config),)
+  peft_model = PeftModel.from_pretrained(model, "yamete4/codegen-350M-mono-QLoRa-flytech")