SDXL-sharding

training script to shard SDXL model across multiple devices

Name		Name	Last commit message	Last commit date
Latest commit History 33 Commits
basket		basket
models		models
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
clip_sharding_layout.json		clip_sharding_layout.json
flattened_clip_xl_sharding.json		flattened_clip_xl_sharding.json
flattened_openclip_xl_sharding.json		flattened_openclip_xl_sharding.json
flattened_unet_xl_sharding.json		flattened_unet_xl_sharding.json
flattened_vae_xl_sharding.json		flattened_vae_xl_sharding.json
inference-xl.py		inference-xl.py
inference1-5.py		inference1-5.py
init.sh		init.sh
partition_pattern.py		partition_pattern.py
training_sharded_1-5.py		training_sharded_1-5.py
training_sharded_sdxl.py		training_sharded_sdxl.py
unet_sharding_layout.json		unet_sharding_layout.json
vae_sharding_layout.json		vae_sharding_layout.json

Provide feedback