change default ft_denoising_steps in eval configs to 0 (assume evaluating pre-trained models)

This commit is contained in:
allenzren
2025-02-04 11:51:47 -05:00
committed by Allen Z. Ren
parent fc42865c77
commit 9032d02eae
28 changed files with 55 additions and 55 deletions
+1 -1
View File
@@ -20,8 +20,8 @@ class DiffusionEval(DiffusionModel):
def __init__(
self,
network_path,
ft_denoising_steps, # if running pre-trained model (not fine-tuned), set it to zero; if running fine-tuned model, need to specify the correct number of denoising steps fine-tuned, so that here it knows which model (base or ft) to use for each denoising step
use_ddim=False,
ft_denoising_steps=0, # if running fine-tuned model, need to specify the correct number of denoising steps fine-tuned, so that here it knows which model (base or ft) to use for each denoising step
**kwargs,
):
# do not let base class load model