frankmorales2020
/

Mistral-7B-text-to-sql

Generated from Trainer

Model card Files Files and versions Metrics Training metrics Community

frankmorales2020 commited on Mar 4

Commit

c3c2def

•

1 Parent(s): 8672c1d

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -36,6 +36,7 @@ model = AutoPeftModelForCausalLM.from_pretrained(
   device_map="auto",
   torch_dtype=torch.float16
 )
 tokenizer = AutoTokenizer.from_pretrained(peft_model_id)
 # Load into the pipeline

   device_map="auto",
   torch_dtype=torch.float16
 )
 tokenizer = AutoTokenizer.from_pretrained(peft_model_id)
 # Load into the pipeline