Important

You are viewing the NeMo 2.0 documentation. This release introduces significant changes to the API and a new library, NeMo Run. We are currently porting all features from NeMo 1.0 to 2.0. For documentation on previous versions or features not yet available in 2.0, please refer to the NeMo 24.07 documentation.

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.