PEFT
code
terryyz commited on
Commit
69330a0
1 Parent(s): 9bbfa63

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