Hugging Face's logo

bigcode
/
astraios-3b-ptuning

PEFT
code
Model card Files Community
1
terryyz commited on
Commit
dd34856
•
1 Parent(s): 1654883

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-3b-ptuning
104
  checkpoint = "bigcode/starcoderbase-3b"
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-3b-ptuning"
104
  checkpoint = "bigcode/starcoderbase-3b"
105
  model = AutoModelForCausalLM.from_pretrained(checkpoint)
106
  model = PeftModel.from_pretrained(model, peft_checkpoint)