# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("rockmiin/ml-codeparrot")
model = AutoModelForCausalLM.from_pretrained("rockmiin/ml-codeparrot", device_map="auto")Quick Links
Pretraining with BigQuery Open Github Dataset using Machine Learning Library (torch, transformers, ..., sklearn)
- Downloads last month
- 12
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="rockmiin/ml-codeparrot")