--- datasets: - HuggingFaceFW/fineweb language: - en library_name: transformers license: unlicense metrics: - accuracy pipeline_tag: text-generation --- # Judge-GPT2 Judge-GPT2 is a custom GPT-2 model designed for text generation tasks. ## Model Description This model is a variant of the GPT-2 model with custom configurations. It has been pretrained and finetuned for specific text generation tasks. ## Usage Here is an example of how to use the model for text generation: ```python from transformers import GPT2Tokenizer, GPT2LMHeadModel tokenizer = GPT2Tokenizer.from_pretrained("Wonder-Griffin/Judge-GPT2") model = GPT2LMHeadModel.from_pretrained("Wonder-Griffin/Judge-GPT2") prompt = "Once upon a time" inputs = tokenizer(prompt, return_tensors="pt") outputs = model.generate(inputs['input_ids'], max_length=50, pad_token_id=tokenizer.eos_token_id) print(tokenizer.decode(outputs[0], skip_special_tokens=True))