Skip to content

Commit b1ef8b1

Browse files
authored
Update README.md
1 parent 5d9a07b commit b1ef8b1

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

README.md

+3
Original file line numberDiff line numberDiff line change
@@ -165,6 +165,9 @@ This will generate images conditioned on the segmentation masks in `MASK_FOLDER/
165165
## Additional Options/Config
166166
Our code has further options for training and evaluation; run `python3 main.py --help` for more information. Further settings still can be changed under `class TrainingConfig:` in `training.py` (some of which are exposed as command-line options for `main.py`, and some of which are not).
167167

168+
## Troubleshooting
169+
- Some users have reported a [bug](https://github.com/mazurowski-lab/segmentation-guided-diffusion/issues/11) when the model attempts to save during training and they receive an error of `module 'safetensors' has no attribute 'torch'`. This appears to be an issue with the `diffusers` library itself in some environments, and may be remedied by [this proposed solution](https://github.com/mazurowski-lab/segmentation-guided-diffusion/issues/11#issuecomment-2251890600).
170+
168171
## Synthetic Paired Breast MRI Dataset Release
169172

170173
<img src='https://github.com/mazurowski-lab/segmentation-guided-diffusion/blob/main/figs/teaser_data.png' width='100%'>

0 commit comments

Comments
 (0)