| { |
| "configuration": "mr", |
| "destination_dtype": "float32", |
| "networks": { |
| "unet": { |
| "copied_from_unet": [], |
| "destination_dtype": "float32", |
| "dropped_keys": [], |
| "file": "model.safetensors", |
| "parameter_count": 180500868, |
| "scale_factor": 0.96240234375, |
| "tensor_count": 435, |
| "tensor_prefix": "unet." |
| }, |
| "vae": { |
| "copied_from_unet": [], |
| "destination_dtype": "float32", |
| "dropped_keys": [], |
| "file": "model.safetensors", |
| "parameter_count": 20944897, |
| "tensor_count": 130, |
| "tensor_prefix": "vae." |
| } |
| }, |
| "source": { |
| "monai": { |
| "distribution_version": "1.6.0", |
| "path": "third_party/monai/MONAI", |
| "revision": "eccefc57550b111ed781d82249dfe77872a0e918" |
| }, |
| "nv_generate_ctmr": { |
| "path": "third_party/nv_generate_ctmr/NV-Generate-CTMR", |
| "revision": "5cb04e82fed71f2fe64a2617ab695be1f5a37fea" |
| }, |
| "weights": { |
| "checkpoints": { |
| "unet": { |
| "filename": "models/diff_unet_3d_rflow-mr.pt", |
| "repo": "nvidia/NV-Generate-MR", |
| "revision": "47a4af30645063be184a322a24ed872c26a9b57f", |
| "url": "https://huggingface.co/nvidia/NV-Generate-MR/resolve/47a4af30645063be184a322a24ed872c26a9b57f/models/diff_unet_3d_rflow-mr.pt" |
| }, |
| "vae": { |
| "filename": "models/autoencoder_v2.pt", |
| "repo": "nvidia/NV-Generate-MR", |
| "revision": "47a4af30645063be184a322a24ed872c26a9b57f", |
| "url": "https://huggingface.co/nvidia/NV-Generate-MR/resolve/47a4af30645063be184a322a24ed872c26a9b57f/models/autoencoder_v2.pt" |
| } |
| }, |
| "repo": "nvidia/NV-Generate-MR", |
| "revision": "47a4af30645063be184a322a24ed872c26a9b57f" |
| } |
| }, |
| "weights_in_git": false |
| } |
|
|
