From 63d5a41997f3ff12da8d3acf4c52fb8db7ccd2d1 Mon Sep 17 00:00:00 2001 From: MotaLee <92512536+ztMotaLee@users.noreply.github.com> Date: Tue, 4 Jun 2024 00:34:11 +0800 Subject: [PATCH] Add files via upload --- configs/dataset/general_inpaint_trainv2.yaml | 1 - ldm/models/diffusion/ddpm_mask.py | 3 +-- model/cldm.py | 3 --- model/cldm_lqcontrol.py | 3 --- 4 files changed, 1 insertion(+), 9 deletions(-) diff --git a/configs/dataset/general_inpaint_trainv2.yaml b/configs/dataset/general_inpaint_trainv2.yaml index 2d8f614..7658f3e 100644 --- a/configs/dataset/general_inpaint_trainv2.yaml +++ b/configs/dataset/general_inpaint_trainv2.yaml @@ -2,7 +2,6 @@ dataset: target: dataset.inpaintingv2.DarkEnhanceDataset params: # Path to the file list. - # file_list: /mnt/petrelfs/liboang/models/hdr_eccv24/DiffBIR-main/filelists/train.list captions_file: path/to/your/file out_size: 512 crop_type: center diff --git a/ldm/models/diffusion/ddpm_mask.py b/ldm/models/diffusion/ddpm_mask.py index 8603721..0fc8aa4 100644 --- a/ldm/models/diffusion/ddpm_mask.py +++ b/ldm/models/diffusion/ddpm_mask.py @@ -566,8 +566,7 @@ def __init__(self, if cond_stage_config == '__is_unconditional__' and not self.force_null_conditioning: conditioning_key = None - # ckpt_path = kwargs.pop("ckpt_path", None) - ckpt_path="/mnt/petrelfs/liboang/models/hdr_eccv24/DiffBIR-main/experiment/lightning_logs/inpaintingv2_final/checkpoints/step=50000.ckpt" + ckpt_path = kwargs.pop("ckpt_path", None) reset_ema = kwargs.pop("reset_ema", False) reset_num_ema_updates = kwargs.pop("reset_num_ema_updates", False) ignore_keys = kwargs.pop("ignore_keys", []) diff --git a/model/cldm.py b/model/cldm.py index 26e5ec3..5e9bdcb 100644 --- a/model/cldm.py +++ b/model/cldm.py @@ -21,9 +21,6 @@ from ldm.modules.distributions.distributions import DiagonalGaussianDistribution from utils.common import frozen_module from .spaced_sampler import SpacedSampler -import sys -sys.path.append("/mnt/petrelfs/liboang/models/hdr_eccv24") -sys.path.append("/mnt/petrelfs/liboang/models/hdr_eccv24/DiffBIR-main") def denormalize(tensor): return tensor * 255.0 diff --git a/model/cldm_lqcontrol.py b/model/cldm_lqcontrol.py index d5517c5..2d786d7 100644 --- a/model/cldm_lqcontrol.py +++ b/model/cldm_lqcontrol.py @@ -20,9 +20,6 @@ from ldm.modules.distributions.distributions import DiagonalGaussianDistribution from utils.common import frozen_module from .spaced_sampler import SpacedSampler -import sys -sys.path.append("/mnt/petrelfs/liboang/models/hdr_eccv24") -sys.path.append("/mnt/petrelfs/liboang/models/hdr_eccv24/DiffBIR-main") def denormalize(tensor): return tensor * 255.0