kd5678 commited on
Commit
93f0fd0
1 Parent(s): b9d747a

Upload paintreward_train_configs.yaml

Browse files
configs/paintreward_train_configs.yaml ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # model configs
2
+ pretrained_model_name: "runwayml/stable-diffusion-inpainting" #"stabilityai/stable-diffusion-2-base"
3
+ seed: 42
4
+
5
+ # reduce memory usage (might slow down training)
6
+ enable_xformers_memory_efficient_attention: True
7
+ enable_vae_tiling: False
8
+ enable_attention_slicing: False
9
+ enable_vae_slicing: False
10
+
11
+ n_inference_steps: 50
12
+ resolution: 512
13
+ guidance_scale: 7.0
14
+ eta: 1.0
15
+
16
+
17
+