Create pornx_trainer.py

pornx_trainer.py

@@ -1,13 +1,8 @@
-from transformers import AutoModelForCausalLM, Trainer, TrainingArguments
+from transformers import AutoModelForCausalLM, Trainer
 from datasets import load_dataset
 
 class PornXTrainer:
     def __init__(self):
         self.repo_id = "ChromiumPlutoniumAI/PornX_AI"
-        self.model_args = {
-            "learning_rate": 2e-5,
-            "num_train_epochs": 3,
-            "per_device_train_batch_size": 4,
-            "gradient_accumulation_steps": 8
-        }
-        
+        self.base_model = "mistral-7b"
+        self.resolution = "4096x2304"