Aduc-sdr commited on
Commit
02f5948
·
verified ·
1 Parent(s): ac5c568

Update inference_configs/inference.yaml

Browse files
Files changed (1) hide show
  1. inference_configs/inference.yaml +6 -6
inference_configs/inference.yaml CHANGED
@@ -11,7 +11,7 @@ use_lora: false
11
  data_config:
12
  eval_batch_size: 8
13
  num_workers: 4
14
- img_size: 544
15
  img_dir: output_bench/ #./ByteMorph-Bench/
16
  grayscale_editing: false
17
  zoom_camera: false
@@ -21,13 +21,13 @@ eval_batch_size: 1
21
  ckpt_dir: ./pretrained_weights/ByteMorpher/dit.safetensors
22
  output_dir: ./test_log/seedmorpher/
23
  logging_dir: logs
24
- mixed_precision: "bf16"
25
  rank: 16
26
  single_blocks: null
27
  double_blocks: null
28
- disable_sampling: calse
29
- sample_width: 544
30
- sample_height: 544
31
- sample_steps: 20
32
  seed: 42
33
  cfg_scale: 2
 
11
  data_config:
12
  eval_batch_size: 8
13
  num_workers: 4
14
+ img_size: 768
15
  img_dir: output_bench/ #./ByteMorph-Bench/
16
  grayscale_editing: false
17
  zoom_camera: false
 
21
  ckpt_dir: ./pretrained_weights/ByteMorpher/dit.safetensors
22
  output_dir: ./test_log/seedmorpher/
23
  logging_dir: logs
24
+ mixed_precision: "fp16"
25
  rank: 16
26
  single_blocks: null
27
  double_blocks: null
28
+ disable_sampling: false
29
+ sample_width: 768
30
+ sample_height: 768
31
+ sample_steps: 50
32
  seed: 42
33
  cfg_scale: 2