lesiondiffusion-mlx / conversion.json
josand's picture
Update v0.2.1 inference API and bundle metadata
643bf83 verified
Raw History Blame Contribute Delete
514 Bytes
{
"schema_version": "medmlx.lesiondiffusion.weights.v1",
"converter_version": 1,
"source_ref": "880736e2b8a4d9becdb6594da7f8af6db19d04d1",
"layout": "source_torch_ncdhw",
"stored_dtype": "float32",
"inference_dtype": {
"lmnet": "float32",
"linet": "float16",
"vq": "float16"
},
"ema": true,
"tensor_file": "model.safetensors",
"tensor_prefixes": {
"linet": "linet.",
"linet_text": "linet_text.",
"lmnet": "lmnet.",
"lmnet_text": "lmnet_text.",
"vq": "vq."
}
}