From 8f42946305742a49a19fdbbd5154b2d93e52faea Mon Sep 17 00:00:00 2001 From: gchapman Date: Sun, 23 Jun 2024 19:44:32 +0100 Subject: [PATCH] Remove override as the new simple loader works it out for us. --- PixArt/loader.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/PixArt/loader.py b/PixArt/loader.py index 0659bca..930568a 100644 --- a/PixArt/loader.py +++ b/PixArt/loader.py @@ -81,8 +81,6 @@ def load_pixart(model_path, model_conf=None): device=model_management.get_torch_device() ) - model_conf.unet_config['depth'] = sum(key.endswith('mlp.fc1.weight') for key in state_dict.keys()) - if model_conf.model_target == "PixArtMS": from .models.PixArtMS import PixArtMS model.diffusion_model = PixArtMS(**model_conf.unet_config)