Update inference_configs/inference.yaml

inference_configs/inference.yaml

@@ -1,7 +1,7 @@
 model_name: "flux-dev"
-use_spatial_condition: false
+use_spatial_condition: true
 share_position_embedding: true
-use_share_weight_referencenet: false 
+use_share_weight_referencenet: false
 use_ip: false
 ip_local_path: null
 ip_repo_id: null
@@ -9,9 +9,9 @@ ip_name: null
 ip_scale: 1.0
 use_lora: false
 data_config:
-  eval_batch_size: 8
-  num_workers: 4
-  img_size: 768
+  eval_batch_size: 1
+  num_workers: 0
+  img_size: 512
   img_dir: output_bench/ #./ByteMorph-Bench/
   grayscale_editing: false
   zoom_camera: false
@@ -21,13 +21,13 @@ eval_batch_size: 1
 ckpt_dir: ./pretrained_weights/ByteMorpher/dit.safetensors
 output_dir: ./test_log/seedmorpher/
 logging_dir: logs
-mixed_precision: "fp16"
+mixed_precision: "bf16"
 rank: 16
 single_blocks: null
 double_blocks: null
 disable_sampling: false
-sample_width: 768
-sample_height: 768
-sample_steps: 50
+sample_width: 512
+sample_height: 512
+sample_steps: 25
 seed: 42
-cfg_scale: 2
+cfg_scale: 3.5