aarohanverma
/

text2sql-flan-t5-base-qlora-finetuned

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

aarohanverma commited on 4 days ago

Commit

5f4b5a6

·

verified ·

1 Parent(s): de0a177

Updated tokenizer

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -212,7 +212,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # Load the fine-tuned model and tokenizer
 model_name = "aarohanverma/text2sql-flan-t5-base-qlora-finetuned"
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name, torch_dtype=torch.bfloat16).to(device)
-tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
 # Ensure decoder start token is set
 if model.config.decoder_start_token_id is None:

 # Load the fine-tuned model and tokenizer
 model_name = "aarohanverma/text2sql-flan-t5-base-qlora-finetuned"
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name, torch_dtype=torch.bfloat16).to(device)
+tokenizer = AutoTokenizer.from_pretrained("aarohanverma/text2sql-flan-t5-base-qlora-finetuned")
 # Ensure decoder start token is set
 if model.config.decoder_start_token_id is None: