rename EditDataPreprocessor to DataPreprocessor

open-mmlab · Jun 1, 2023 · 535016a · 535016a
1 parent 1686e66
commit 535016a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/projects/glide/configs/glide_ddim-classifier-free_laion-64-256.py b/projects/glide/configs/glide_ddim-classifier-free_laion-64-256.py
@@ -54,8 +54,7 @@
 
 model = dict(
     type='Glide',
-    data_preprocessor=dict(
-        type='EditDataPreprocessor', mean=[127.5], std=[127.5]),
+    data_preprocessor=dict(type='DataPreprocessor', mean=[127.5], std=[127.5]),
     unet=unet_cfg,
     diffusion_scheduler=dict(
         type='EditDDIMScheduler',

diff --git a/projects/glide/models/glide.py b/projects/glide/models/glide.py
@@ -69,7 +69,7 @@ def __init__(self,
                  classifier: Optional[dict] = None,
                  classifier_scale: float = 1.0,
                  data_preprocessor: Optional[ModelType] = dict(
-                     type='EditDataPreprocessor'),
+                     type='DataPreprocessor'),
                  pretrained_cfgs: Optional[dict] = None):
 
         super().__init__(data_preprocessor=data_preprocessor)