pixart-alpha

e2364931 · mashun1 · e2364931 · e2364931 · e2364931 · e2364931
Commit e2364931 authored Apr 27, 2024 by mashun1
20 changed files
--- a/diffusion/model/hed.py
+++ b/diffusion/model/hed.py
--- a/diffusion/model/llava/__init__.py
+++ b/diffusion/model/llava/__init__.py
+from diffusion.model.llava.llava_mpt import LlavaMPTForCausalLM, LlavaMPTConfig
\ No newline at end of file
--- a/diffusion/model/llava/llava_mpt.py
+++ b/diffusion/model/llava/llava_mpt.py
--- a/diffusion/model/llava/mpt/attention.py
+++ b/diffusion/model/llava/mpt/attention.py
--- a/diffusion/model/llava/mpt/blocks.py
+++ b/diffusion/model/llava/mpt/blocks.py
--- a/diffusion/model/llava/mpt/configuration_mpt.py
+++ b/diffusion/model/llava/mpt/configuration_mpt.py
--- a/diffusion/model/llava/mpt/modeling_mpt.py
+++ b/diffusion/model/llava/mpt/modeling_mpt.py
--- a/diffusion/model/llava/mpt/norm.py
+++ b/diffusion/model/llava/mpt/norm.py
--- a/diffusion/model/llava/mpt/param_init_fns.py
+++ b/diffusion/model/llava/mpt/param_init_fns.py
--- a/diffusion/model/nets/PixArt.py
+++ b/diffusion/model/nets/PixArt.py
--- a/diffusion/model/nets/PixArtMS.py
+++ b/diffusion/model/nets/PixArtMS.py
--- a/diffusion/model/nets/PixArt_blocks.py
+++ b/diffusion/model/nets/PixArt_blocks.py
--- a/diffusion/model/nets/__init__.py
+++ b/diffusion/model/nets/__init__.py
+from .PixArt import PixArt, PixArt_XL_2
+from .PixArtMS import PixArtMS, PixArtMS_XL_2, PixArtMSBlock
+from .pixart_controlnet import ControlPixArtHalf, ControlPixArtMSHalf
\ No newline at end of file
--- a/diffusion/model/nets/pixart_controlnet.py
+++ b/diffusion/model/nets/pixart_controlnet.py
--- a/diffusion/model/respace.py
+++ b/diffusion/model/respace.py
--- a/diffusion/model/sa_solver.py
+++ b/diffusion/model/sa_solver.py
--- a/diffusion/model/t5.py
+++ b/diffusion/model/t5.py
--- a/diffusion/model/timestep_sampler.py
+++ b/diffusion/model/timestep_sampler.py
--- a/diffusion/model/utils.py
+++ b/diffusion/model/utils.py
--- a/diffusion/sa_sampler.py
+++ b/diffusion/sa_sampler.py