SQLCreator

Model Overview

This model is designed to generate SQL queries based on input prompts. It is based on GPT-2 and trained with custom datasets.

Usage

To use this model, follow these steps:

  1. Install the necessary libraries:
    pip install transformers
    
  2. Load the model and tokenizer:
    from transformers import AutoModelForCausalLM, AutoTokenizer
    
    model_name = “Kasivs/SQLCreator"
    tokenizer = AutoTokenizer.from_pretrained(model_name)
    model = AutoModelForCausalLM.from_pretrained(model_name)
    
    # Example usage
    inputs = tokenizer("SELECT * FROM users WHERE", return_tensors="pt")
    outputs = model.generate(inputs["input_ids"])
    print(tokenizer.decode(outputs[0]))
    

Training

Details about how the model was trained, including dataset information and training parameters.

License

Specify the license under which the model is distributed.

Citation

Provide citation information if applicable.

Contact

Your contact information or any related resources.

Downloads last month
0
Safetensors
Model size
86.2M params
Tensor type
F32
·
Inference Providers NEW
This model is not currently available via any of the supported third-party Inference Providers, and the HF Inference API does not support flair models with pipeline type text2text-generation

Dataset used to train Kasivs/SQLCreator