Fine-Tuned LLM for Code Generation

This model is a fine-tuned version of GPT-2 for generating Java code. It was trained on a subset of the WikiText dataset to improve its ability to generate code snippets.

How to Use

You can use this model with the transformers library:

from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = "14harshaldhote/plutusLLM"  # Replace with your model name
model = AutoModelForCausalLM.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

prompt = "Write a Python function to compute factorial"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
Downloads last month
17
Safetensors
Model size
81.9M params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support