Post
18
if you are interested in HUB (https://saemi410.github.io/HUB/) and you want to run the UCE (https://unified.baulab.info), please check:
- Model weights for UCE here: telcom/uce_NSFW
- datasets and more download materials from: telcom/HUB_reference_dataset
Please note that: The weights here are a partial UNet state_dict (e.g. only a few attn2.to_k/to_v weights). So in loading it with PyTorch’s default strict=False is required.
- Model weights for UCE here: telcom/uce_NSFW
- datasets and more download materials from: telcom/HUB_reference_dataset
Please note that: The weights here are a partial UNet state_dict (e.g. only a few attn2.to_k/to_v weights). So in loading it with PyTorch’s default strict=False is required.