Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -100,7 +100,7 @@ Answer:"
|
|
100 |
from peft import PeftModel
|
101 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
102 |
|
103 |
-
peft_checkpoint = bigcode/astraios-1b-adapterh
|
104 |
checkpoint = "bigcode/starcoderbase-1b"
|
105 |
model = AutoModelForCausalLM.from_pretrained(checkpoint)
|
106 |
model = PeftModel.from_pretrained(model, peft_checkpoint)
|
|
|
100 |
from peft import PeftModel
|
101 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
102 |
|
103 |
+
peft_checkpoint = "bigcode/astraios-1b-adapterh"
|
104 |
checkpoint = "bigcode/starcoderbase-1b"
|
105 |
model = AutoModelForCausalLM.from_pretrained(checkpoint)
|
106 |
model = PeftModel.from_pretrained(model, peft_checkpoint)
|