Add files via upload

ztMotaLee · Jun 3, 2024 · 63d5a41 · 63d5a41
1 parent e3f0a95
commit 63d5a41
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 9 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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