mlx-generate-ctmr / ct /conversion.json
josand's picture
Bundle paired-CT label mapping and anatomy sizes; update conversion notes; add quick start
fa7fbef verified
Raw History Blame Contribute Delete
2.4 kB
{
"configuration": "ct",
"destination_dtype": "float32",
"networks": {
"controlnet": {
"copied_from_unet": [],
"destination_dtype": "float32",
"dropped_keys": [],
"file": "model.safetensors",
"parameter_count": 72037072,
"tensor_count": 231,
"tensor_prefix": "controlnet."
},
"unet": {
"copied_from_unet": [],
"destination_dtype": "float32",
"dropped_keys": [],
"file": "model.safetensors",
"parameter_count": 180500868,
"scale_factor": 1.0311251878738403,
"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": {
"controlnet": {
"filename": "models/controlnet_3d_rflow-ct.pt",
"repo": "nvidia/NV-Generate-CT",
"revision": "48a68e95c77a095870e3374f5b2cca1dd211f317",
"url": "https://huggingface.co/nvidia/NV-Generate-CT/resolve/48a68e95c77a095870e3374f5b2cca1dd211f317/models/controlnet_3d_rflow-ct.pt"
},
"unet": {
"filename": "models/diff_unet_3d_rflow-ct.pt",
"repo": "nvidia/NV-Generate-CT",
"revision": "48a68e95c77a095870e3374f5b2cca1dd211f317",
"url": "https://huggingface.co/nvidia/NV-Generate-CT/resolve/48a68e95c77a095870e3374f5b2cca1dd211f317/models/diff_unet_3d_rflow-ct.pt"
},
"vae": {
"filename": "models/autoencoder_v1.pt",
"repo": "nvidia/NV-Generate-CT",
"revision": "48a68e95c77a095870e3374f5b2cca1dd211f317",
"url": "https://huggingface.co/nvidia/NV-Generate-CT/resolve/48a68e95c77a095870e3374f5b2cca1dd211f317/models/autoencoder_v1.pt"
}
},
"repo": "nvidia/NV-Generate-CT",
"revision": "48a68e95c77a095870e3374f5b2cca1dd211f317"
}
},
"weights_in_git": false
}