File size: 265 Bytes
2cc02de
33d6de5
 
 
 
 
2cc02de
 
1
2
3
4
5
6
7
8
9
from transformers import AutoModelForCausalLM, Trainer
from datasets import load_dataset

class PornXTrainer:
    def __init__(self):
        self.repo_id = "ChromiumPlutoniumAI/PornX_AI"
        self.base_model = "mistral-7b"
        self.resolution = "4096x2304"