Important

NeMo 2.0 is an experimental feature and currently released in the dev container only: nvcr.io/nvidia/nemo:dev. Please refer to NeMo 2.0 overview for information on getting started.

Framework Inference

For NSFW models, the inference script generates image scores for NSFW content. Values closer to 1 are strong NSFW while -1 are strong safe.

To enable inference stage with a NSFW model, configure the configuration files:

In the defaults section of conf/config.yaml, update the fw_inference field to point to the desired NSFW inference configuration file. For example, if you want to use the nsfw/nsfw.yaml configuration, change the fw_inference field to nsfw/nsfw.
```
defaults:
  - fw_inference: nsfw/nsfw
  ...
```
In the stages field of conf/config.yaml, make sure the fw_inference stage is included. For example,
```
stages:
  - fw_inference
  ...
```
Configure image_path field of conf/fw_inference/nsfw/nsfw.yaml.
Execute the launcher pipeline: python3 main.py.