From fec9cbfadda6cdc45f73ee24ffc462dba9815b36 Mon Sep 17 00:00:00 2001
From: aiXander <xandersteenbruggex@gmail.com>
Date: Tue, 14 May 2024 15:11:07 -0700
Subject: [PATCH] fix cog predict

---
 cog.yaml                      | 27 ++++-----------------------
 requirements.txt              | 30 +++++++++++++-----------------
 training_args_object.json     |  2 +-
 training_args_style_sd15.json | 10 +++++-----
 4 files changed, 23 insertions(+), 46 deletions(-)

diff --git a/cog.yaml b/cog.yaml
index d6ea325..e284eba 100755
--- a/cog.yaml
+++ b/cog.yaml
@@ -6,33 +6,14 @@ build:
   cuda: "11.8"
   python_version: "3.9"
   system_packages:
-    - "libgl1-mesa-glx"
     - "ffmpeg"
+    - "libgl1-mesa-glx"
+    - "libegl1-mesa-dev"
     - "libsm6"
     - "libxext6"
-  python_packages:
-    - ujson
-    - "scipy==1.11.1"
-    - "diffusers==0.27.2"
-    - "peft==0.10.0"
-    - "torch==2.2.1"
-    - "transformers==4.39.3"
-    - "invisible-watermark==0.2.0"
-    - "accelerate==0.21.0"
-    - "pandas==2.0.3"
-    - "torchvision==0.17.1"
-    - "numpy==1.25.1"
-    - "fire==0.5.0"
-    - "opencv-python>=4.1.0.25"
-    - "mediapipe==0.10.2"
-    - "openai==1.6.1"
-    - python-dotenv
-    - prodigyopt
-    - omegaconf
-
+  python_requirements: requirements.txt
   run:
-    - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.0.1/pget" && chmod +x /usr/local/bin/pget
     - wget http://thegiflibrary.tumblr.com/post/11565547760 -O face_landmarker_v2_with_blendshapes.task -q https://storage.googleapis.com/mediapipe-models/face_landmarker/face_landmarker/float16/1/face_landmarker.task
 
 predict: "predict.py:Predictor"
-image: "r8.im/abraham-ai/sdxl-lora-trainer"
+image: "r8.im/edenartlab/sdxl-lora-trainer"
diff --git a/requirements.txt b/requirements.txt
index 50db0d9..8ff19df 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,21 +1,17 @@
-scipy
-ujson
-diffusers==0.25.1
-peft==0.9.0
-torch==2.0.1
-transformers==4.31.0
+torch==2.1.0
+torchvision==0.16.0
+transformers==4.38.1
+diffusers==0.27.2
+ujson==5.9.0
+scipy==1.12.0
+peft==0.10.0
 invisible-watermark==0.2.0
-accelerate==0.21.0
-pandas==2.0.3
-torchvision==0.15.2
-numpy==1.25.1
-pandas==2.0.3
-fire==0.5.0
+accelerate==0.27.2
+pandas==2.2.1
+numpy==1.26.4
 opencv-python>=4.1.0.25
-mediapipe==0.10.2
-openai==1.2.4
+mediapipe==0.10.11
+openai==1.14.0
 python-dotenv
 prodigyopt
-omegaconf
-ujson
-py7zr
\ No newline at end of file
+omegaconf
\ No newline at end of file
diff --git a/training_args_object.json b/training_args_object.json
index cfee244..799279f 100644
--- a/training_args_object.json
+++ b/training_args_object.json
@@ -15,7 +15,7 @@
     "n_tokens": 2,
     "ti_lr": 0.001,
     "ti_weight_decay": 0.0005,
-    "text_encoder_lora_optimizer": "adamw",
+    "text_encoder_lora_optimizer": null,
     "text_encoder_lora_lr": 1.0e-4,
     "text_encoder_lora_weight_decay": 1e-5,
     "text_encoder_lora_rank": 12,
diff --git a/training_args_style_sd15.json b/training_args_style_sd15.json
index 654b15d..62cab56 100644
--- a/training_args_style_sd15.json
+++ b/training_args_style_sd15.json
@@ -1,15 +1,15 @@
 {
-    "output_dir": "lora_models/does",
-    "sd_model_version": "sd15",
-    "lora_training_urls": "/home/rednax/Documents/datasets/does",
+    "output_dir": "lora_models/style",
+    "sd_model_version": "sdxl",
+    "lora_training_urls": "/data/xander/Projects/cog/GitHub_repos/diffusion_trainer2/datasets/eden_crystals",
     "concept_mode": "style",
     "seed": 0,
     "resolution": 512,
     "train_batch_size": 4,
     "n_sample_imgs": 6,
-    "max_train_steps": 600,
+    "max_train_steps": 500,
     "token_warmup_steps": 0,
-    "checkpointing_steps": 100,
+    "checkpointing_steps": 200,
     "gradient_accumulation_steps": 1,
     "n_tokens": 2,
     "ti_lr": 0.001,