Hugging Face's logo

szalontaib
/
MegaBugInject

PEFT
Model card Files
xet
 Community
1
szalontaib commited on
Commit
d955ffe
·
verified ·
1 Parent(s): 0a65e4e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -27,7 +27,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
27
  from peft import PeftModel
28
 
29
  model_id_pretrained = 'WizardLMTeam/WizardCoder-Python-13B-V1.0'
30
- model_id_finetuned = 'szalontaib/MegaDiffInject'
31
 
32
  tokenizer = AutoTokenizer.from_pretrained(model_id_pretrained, add_eos_token=False)
33
  model = AutoModelForCausalLM.from_pretrained(model_id_pretrained, device_map='auto', dtype=torch.float16, trust_remote_code=True)
 
27
  from peft import PeftModel
28
 
29
  model_id_pretrained = 'WizardLMTeam/WizardCoder-Python-13B-V1.0'
30
+ model_id_finetuned = 'szalontaib/MegaBugInject'
31
 
32
  tokenizer = AutoTokenizer.from_pretrained(model_id_pretrained, add_eos_token=False)
33
  model = AutoModelForCausalLM.from_pretrained(model_id_pretrained, device_map='auto', dtype=torch.float16, trust_remote_code=True)