Hugging Face's logo

bigcode
/
astraios-1b-lora

PEFT
code
Model card Files Community
1
terryyz commited on
Commit
9d0dcc9
•
1 Parent(s): e3a45fb

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