PEFT
code
terryyz commited on
Commit
c4e6204
1 Parent(s): 1998e33

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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)