File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change 27
27
# cd output_folder/dit/overfitting_depth_16_token_4096_lr1e4/ckpt/ckpt-step=00004000.ckpt
28
28
# python3 zero_to_fp32.py ./ ./out --max_shard_size 30GB
29
29
# then you can get output_folder/dit/overfitting_depth_16_token_4096_lr1e4/ckpt/ckpt-step=00004000.ckpt/out/pytorch_model.bin
30
- ckpt_cfg_path = 'output_folder/dit/overfitting_depth_16_token_4096_lr1e4_uc /hunyuandit-mini-overfitting-flowmatching-dinol518-bf16-lr1e4-4096.yaml'
30
+ ckpt_cfg_path = 'output_folder/dit/overfitting_depth_16_token_4096_lr1e4 /hunyuandit-mini-overfitting-flowmatching-dinol518-bf16-lr1e4-4096.yaml'
31
31
ckpt_path = 'output_folder/dit/overfitting_depth_16_token_4096_lr1e4/ckpt/ckpt-step=00004000.ckpt/out/pytorch_model.bin'
32
32
config = yaml .safe_load (open (ckpt_cfg_path , 'r' ))
33
33
model = instantiate_from_config (config ['model' ]['params' ]['denoiser_cfg' ])
You can’t perform that action at this time.
0 commit comments