fix the sample num bug in inference.

Signed-off-by: lawrence-cj <[email protected]>
NVlabs · Dec 22, 2024 · 295ee05 · 295ee05
1 parent 655c2f7
commit 295ee05
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/scripts/bash_run_inference_metric_geneval.sh b/scripts/bash_run_inference_metric_geneval.sh
@@ -12,7 +12,7 @@ fi
 # ============ end of custom code block ===========
 
 default_step=20
-default_sample_nums=533
+default_sample_nums=553
 default_sampling_algo="flow_dpm-solver"
 default_add_label=''
 

diff --git a/scripts/infer_metric_run_inference_metric_geneval.sh b/scripts/infer_metric_run_inference_metric_geneval.sh
@@ -5,7 +5,7 @@ output_dir=output
 # ============ 0. start of custom code block ============
 #### Infer Hyper
 default_step=20                             # inference step for diffusion model
-default_sample_nums=533                   # inference first $sample_nums sample in list(json.keys())
+default_sample_nums=553                   # inference first $sample_nums sample in list(json.keys())
 default_sampling_algo="flow_dpm-solver"
 default_add_label=''
 default_log_geneval=false

diff --git a/scripts/infer_run_inference_geneval.sh b/scripts/infer_run_inference_geneval.sh
@@ -7,7 +7,7 @@
 # ================= sampler & data =================
 np=8    # number of GPU to use
 default_step=20   # 14
-default_sample_nums=533
+default_sample_nums=553
 default_sampling_algo="flow_dpm-solver"
 default_add_label=''
 

diff --git a/scripts/infer_run_inference_geneval_diffusers.sh b/scripts/infer_run_inference_geneval_diffusers.sh
@@ -2,7 +2,7 @@
 # ================= sampler & data =================
 np=8    # number of GPU to use
 default_step=20   # 14
-default_sample_nums=533
+default_sample_nums=553
 default_sampling_algo="dpm-solver"
 default_add_label=''
 

diff --git a/scripts/inference_geneval.py b/scripts/inference_geneval.py
@@ -346,7 +346,7 @@ class SanaInference(SanaConfig):
     batch_size: int = field(default=1, metadata={"help": "how many samples can be produced simultaneously"})
     skip_grid: bool = field(default=False, metadata={"help": "skip saving grid"})
     model_path: Optional[str] = field(default=None, metadata={"help": "Path to the model file (optional)"})
-    sample_nums: int = 533
+    sample_nums: int = 553
     cfg_scale: float = 4.5
     pag_scale: float = 1.0
     sampling_algo: str = field(

diff --git a/scripts/inference_geneval_diffusers.py b/scripts/inference_geneval_diffusers.py
@@ -151,7 +151,7 @@ def parse_args():
         help="skip saving grid",
     )
 
-    parser.add_argument("--sample_nums", default=533, type=int)
+    parser.add_argument("--sample_nums", default=553, type=int)
     parser.add_argument("--add_label", default="", type=str)
     parser.add_argument("--exist_time_prefix", default="", type=str)
     parser.add_argument("--gpu_id", type=int, default=0)

diff --git a/tools/metrics/compute_geneval.sh b/tools/metrics/compute_geneval.sh
@@ -6,7 +6,7 @@ geneval=true
 
 np=8    # number of GPU to use
 py=tools/metrics/geneval/evaluation/evaluate_images.py
-default_sample_nums=533
+default_sample_nums=553
 report_to=wandb
 default_log_suffix_label=''