SynLayers commited on
Commit
209d166
·
verified ·
1 Parent(s): 130c3c9

Upload infer/infer.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. infer/infer.yaml +9 -34
infer/infer.yaml CHANGED
@@ -7,55 +7,30 @@ source_size: 1024
7
  target_size: 1024
8
 
9
  # Real-world inference defaults
10
- data_dir: "/project/llmsvgen/share/data/kmw_layered_dataset/real_world_inference"
11
- image_dir: "/project/llmsvgen/share/data/kmw_layered_dataset/real_world_inference/layers_real_test_1024"
12
- test_jsonl: "/project/llmsvgen/share/data/kmw_layered_dataset/real_world_inference/caption_bbox_infer.jsonl"
13
 
14
  # Model paths
15
- pretrained_model_name_or_path: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_checkpoints/FLUX.1-dev"
16
- pretrained_adapter_path: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_checkpoints/FLUX.1-dev-Controlnet-Inpainting-Alpha"
17
  transp_vae_path: "ckpt/trans_vae/0008000.pt"
18
 
19
  # Pre-trained LoRA weights
20
  pretrained_lora_dir: "ckpt/pre_trained_LoRA"
21
  artplus_lora_dir: "ckpt/prism_ft_LoRA"
22
 
23
- # below is for 18k dataset
24
- #lora_ckpt: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_18k/step_90000/transformer"
25
- #layer_ckpt: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_18k/step_90000"
26
- #adapter_lora_dir: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_18k/step_90000/adapter"
27
-
28
- # below is for 20k dataset
29
- #lora_ckpt: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_20k/step_120000/transformer"
30
- #layer_ckpt: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_20k/step_120000"
31
- #adapter_lora_dir: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_20k/step_120000/adapter"
32
-
33
- # below is for 30k dataset
34
- #lora_ckpt: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_30k/step_150000/transformer"
35
- #layer_ckpt: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_30k/step_150000"
36
- #adapter_lora_dir: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_30k/step_150000/adapter"
37
-
38
- # below is for 40k dataset
39
- #lora_ckpt: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_40k/step_250000/transformer"
40
- #layer_ckpt: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_40k/step_250000"
41
- #adapter_lora_dir: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_40k/step_250000/adapter"
42
-
43
- # below is for 50k dataset
44
- #lora_ckpt: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_50k/step_200000/transformer"
45
- #layer_ckpt: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_50k/step_200000"
46
- #adapter_lora_dir: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_train_dataset/ckpt_prism_scaleup_1024_50k/step_200000/adapter"
47
-
48
  # unified real-world decomposition checkpoint
49
- lora_ckpt: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_ckpt/step_120000/transformer"
50
- layer_ckpt: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_ckpt/step_120000"
51
- adapter_lora_dir: "/project/llmsvgen/share/data/kmw_layered_checkpoint/SynLayers_ckpt/step_120000/adapter"
52
 
53
  # Inference settings
54
  cfg: 4.0
55
  adapter_scale: 0.9
56
  max_sequence_length: 1024
57
 
58
- save_dir: "/project/llmsvgen/share/data/kmw_layered_dataset/real_world_inference/results"
59
  #run_name: "step_120000" # optional manual override
60
 
61
  # Sample range control (1-based indexing)
 
7
  target_size: 1024
8
 
9
  # Real-world inference defaults
10
+ data_dir: "demo/outputs/real_world_demo"
11
+ image_dir: "demo/outputs/real_world_demo/layers_real_test_1024"
12
+ test_jsonl: "demo/outputs/real_world_demo/caption_bbox_infer.jsonl"
13
 
14
  # Model paths
15
+ pretrained_model_name_or_path: "SynLayers_checkpoints/FLUX.1-dev"
16
+ pretrained_adapter_path: "SynLayers_checkpoints/FLUX.1-dev-Controlnet-Inpainting-Alpha"
17
  transp_vae_path: "ckpt/trans_vae/0008000.pt"
18
 
19
  # Pre-trained LoRA weights
20
  pretrained_lora_dir: "ckpt/pre_trained_LoRA"
21
  artplus_lora_dir: "ckpt/prism_ft_LoRA"
22
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
23
  # unified real-world decomposition checkpoint
24
+ lora_ckpt: "SynLayers_ckpt/step_120000/transformer"
25
+ layer_ckpt: "SynLayers_ckpt/step_120000"
26
+ adapter_lora_dir: "SynLayers_ckpt/step_120000/adapter"
27
 
28
  # Inference settings
29
  cfg: 4.0
30
  adapter_scale: 0.9
31
  max_sequence_length: 1024
32
 
33
+ save_dir: "demo/outputs/real_world_demo/results"
34
  #run_name: "step_120000" # optional manual override
35
 
36
  # Sample range control (1-based indexing)