File size: 924 Bytes
356f778 47d213b 356f778 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
---
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)) |