/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
examples/images/dreambooth/debug.py
21 строка
440 B
Hongxin Liu
[misc] refactor launch API and tensor constructor (#5666)
29 апр 2024, 05:40
Не верифицирован
29 апр 2024, 05:40
7f8b166
Код
Авторство
О чём код?
""" torchrun --standalone --nproc_per_node=1 debug.py """ from diffusers import AutoencoderKL import colossalai from colossalai.zero import ColoInitContext path = "/data/scratch/diffuser/stable-diffusion-v1-4" colossalai.launch_from_torch() with ColoInitContext(device="cpu"): vae = AutoencoderKL.from_pretrained( path, subfolder="vae", revision=None, ) for n, p in vae.named_parameters(): print(n)