Hugging Face's logo

bigcode
/
astraios-1b-parallel

PEFT
code
Model card Files
xet
Community
1
terryyz commited on
Commit
a8ee4a0
·
1 Parent(s): bf6fd7b

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