search

PixArt-alpha / PixArt-sigma

PixArt-Σ: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation
https://pixart-alpha.github.io/PixArt-sigma-project/
GNU Affero General Public License v3.0
1.47k stars 70 forks source link

DMD load problems #26

Closed chuck-ma closed 3 months ago

chuck-ma commented 3 months ago

First ,it seems that the model should be loaded as safetensors.Because no safetensors files in https://huggingface.co/PixArt-alpha/PixArt-Alpha-DMD-XL-2-512x512/tree/main.

pipe = PixArtAlphaPipeline.from_pretrained( model_path, subfolder="transformer", torch_dtype=weight_dtype, use_safetensors=False, )

Besides, it seems that I can't load the PixArtAlphaPipeline somehow cause some config file missing.

OSError: Error no file named config.json found in directory /root/.cache/huggingface/hub/models--PixArt-alpha--PixArt-Alpha-DMD-XL-2-512x512/snapshots/57df7bb32daad69bc1d6c825a275935113802169.

EYcab commented 3 months ago

The same error caught here@lawrence-cj

lawrence-cj commented 3 months ago

Fixed. Let's have a try. https://github.com/PixArt-alpha/PixArt-sigma/commit/2ba2f6efd4b5c6e922ecda04102ceba1430003ec