mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-02 17:46:42 +00:00
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
This commit is contained in:
@@ -1,16 +1,16 @@
|
||||
'''
|
||||
"""
|
||||
torchrun --standalone --nproc_per_node=1 debug.py
|
||||
'''
|
||||
"""
|
||||
|
||||
from diffusers import AutoencoderKL
|
||||
|
||||
import colossalai
|
||||
from colossalai.zero import ColoInitContext, post_process_colo_init_ctx
|
||||
from colossalai.zero import ColoInitContext
|
||||
|
||||
path = "/data/scratch/diffuser/stable-diffusion-v1-4"
|
||||
|
||||
colossalai.launch_from_torch(config={})
|
||||
with ColoInitContext(device='cpu'):
|
||||
with ColoInitContext(device="cpu"):
|
||||
vae = AutoencoderKL.from_pretrained(
|
||||
path,
|
||||
subfolder="vae",
|
||||
|
Reference in New Issue
Block a user