Spaces:
Paused
Paused
Update inference_configs/inference.yaml
Browse files
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:
|
| 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: "
|
| 25 |
rank: 16
|
| 26 |
single_blocks: null
|
| 27 |
double_blocks: null
|
| 28 |
-
disable_sampling:
|
| 29 |
-
sample_width:
|
| 30 |
-
sample_height:
|
| 31 |
-
sample_steps:
|
| 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
|