concept-ablation

Running on T4

nupurkmr9 commited on Nov 25, 2023

Commit

6ee394b

•

1 Parent(s): 8d26fb8

Update trainer.py

Files changed (1) hide show

trainer.py CHANGED Viewed

@@ -100,7 +100,7 @@ def train_submit(
           --scale_lr --hflip \
           --parameter_group {parameter_group} \
           --openai_key "{openai_key}" \
-          --enable_xformers_memory_efficient_attention  --num_class_images 1000
         '''
         if concept_type == 'style':

           --scale_lr --hflip \
           --parameter_group {parameter_group} \
           --openai_key "{openai_key}" \
+          --enable_xformers_memory_efficient_attention  --num_class_images 500
         '''
         if concept_type == 'style':