stabvle diffusion 3 running for inference

natural images only
Likalto4 · Jun 17, 2024 · fdf4705 · fdf4705
1 parent 94747fc
commit fdf4705
Show file tree

Hide file tree

Showing 3 changed files with 109 additions and 3 deletions.
diff --git a/experiments/sd3_mame/trying_sd3.ipynb b/experiments/sd3_mame/trying_sd3.ipynb
diff --git a/experiments/standard_mame/config_file.yaml b/experiments/standard_mame/config_file.yaml
@@ -8,7 +8,7 @@ revision: fp16 # str # type of model kept for fp16 optimization
 tokenizer_name: null # str
 # data:
 instance_data_dir: /home/ricardo/Documents/PhD/projects/diffusion-models_master/data/images/breast40k_RGB_healthy # Mandatory, str
-output_dir: /home/ricardo/Documents/PhD/projects/diffusion-models_master/results/sd-v2-base_OPTIMAM # Mandatory, str <------------
+output_dir: /home/ricardo/Documents/PhD/projects/diffusion-models_master/results/sd-v2-base_OPTIMAM_256 # Mandatory, str <------------
 # hub:
 push_to_hub: true
 hub_token: null
@@ -40,7 +40,7 @@ resolution: 512
 center_crop: false # center or random crop # this is done independenly as preprocessing step
 train_text_encoder: true # train the text encoder
 train_batch_size: 16 # per device
-gradient_accumulation_steps: 4
+gradient_accumulation_steps: 16
 learning_rate: 1.0e-6 # lr after (potential) warmup
 scale_lr: False #Scale the learning rate by the number of GPUs, gradient accumulation steps, and batch size
 lr_scheduler: constant

diff --git a/experiments/standard_mame/dreambooth_mammo_prompt.py b/experiments/standard_mame/dreambooth_mammo_prompt.py
@@ -14,7 +14,7 @@
 exp_path = Path.cwd().resolve() # experiment path
 # visible GPUs
 os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
-os.environ["CUDA_VISIBLE_DEVICES"]="0"
+os.environ["CUDA_VISIBLE_DEVICES"]="1"
 
 import argparse
 import yaml