sisinflab-ai
/

GradeSQL-32B-ORM-Spider

Model card Files Files and versions

MattiaTritto commited on 15 days ago

Commit

e2335de

·

verified ·

1 Parent(s): 1c1ba94

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -79,7 +79,7 @@ ORDER BY SUM(Total_Horses) ASC;
 Is the SQL correct?"""
 base_model = AutoModelForCausalLM.from_pretrained("seeklhy/OmniSQL-32B", torch_dtype="auto", device_map="auto")
-peft_model = PeftModel.from_pretrained(base_model, "sisinflab-ai/GradeSQL-32B-ORM-Bird")
 orm_model = peft_model.merge_and_unload()
 orm_tokenizer = AutoTokenizer.from_pretrained("seeklhy/OmniSQL-32B", use_fast=True)

 Is the SQL correct?"""
 base_model = AutoModelForCausalLM.from_pretrained("seeklhy/OmniSQL-32B", torch_dtype="auto", device_map="auto")
+peft_model = PeftModel.from_pretrained(base_model, "sisinflab-ai/GradeSQL-32B-ORM-Spider")
 orm_model = peft_model.merge_and_unload()
 orm_tokenizer = AutoTokenizer.from_pretrained("seeklhy/OmniSQL-32B", use_fast=True)