How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="ankit011/phi-2-code", trust_remote_code=True)
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("ankit011/phi-2-code", trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained("ankit011/phi-2-code", trust_remote_code=True)
Quick Links

Model Details

Fine-Tuned Micrsoft/phi-2 model to generate Python code.

  • Developed by: [Ankit Kumar Mishra]
  • Model type: [Code Generation]
  • Finetuned from model [optional]: [Microsoft/phi-2]
Downloads last month
5
Safetensors
Model size
3B params
Tensor type
F16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train ankit011/phi-2-code