Skip to content

Commit

Permalink
change to 3000
Browse files Browse the repository at this point in the history
Signed-off-by: Can-Zhao <[email protected]>
  • Loading branch information
Can-Zhao committed Mar 8, 2025
2 parents ebec91b + f1bfed3 commit c20b892
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion models/maisi_ct_generative/configs/inference.json
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
"output_dir": "$@bundle_root + '/output'",
"create_output_dir": "$Path(@output_dir).mkdir(exist_ok=True)",
"device": "$torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')",
"trained_autoencoder_path": "$@model_dir + '/autoencoder_epoch273.pt'",
"trained_autoencoder_path": "$@model_dir + '/autoencoder.pt'",
"trained_diffusion_path": "$@model_dir + '/diffusion_unet.pt'",
"trained_controlnet_path": "$@model_dir + '/controlnet.pt'",
"trained_mask_generation_autoencoder_path": "$@model_dir + '/mask_generation_autoencoder.pt'",
Expand Down

0 comments on commit c20b892

Please sign in to comment.