Use native ops for PixArt

This commit is contained in:
City
2024-12-11 00:52:48 +01:00
parent 45423d9673
commit d5a47fa3e5
4 changed files with 157 additions and 157 deletions
+1 -2
View File
@@ -93,8 +93,7 @@ def model_config_from_unet(sd):
config["pe_interpolation"] = 1
model_config = PixArtConfig(config)
model_config.unet_class = model_class
logging.info(f"Detected PixArt model as [{model_class}]")
logging.info(f"PixArt config:\n{config}")
logging.debug(f"PixArt config: {model_class}\n{config}")
return model_config
resolutions = {