Update README.md

README.md

@@ -22,8 +22,7 @@ See more informations at [our GitHub](https://github.com/boostcampaitech6/level2
 ## Quick Tour
 ```python
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
-from peft import PeftModel
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
 model = AutoModelForCausalLM.from_pretrained(
     "EleutherAI/polyglot-ko-5.8b",
@@ -32,10 +31,9 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto",
 )
 model.config.use_cache = True
-
-model = PeftModel.from_pretrained(
-  model, "m2af/EleutherAI-polyglot-ko-5.8b-adapter"
-)
+    
+model.load_adapter("m2af/EleutherAI-polyglot-ko-5.8b-adapter", "loaded")
+model.set_adapter("loaded")
 
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 tokenizer.pad_token = tokenizer.eos_token