Skip to content

Change NSFW Model#307

Merged
ryantwolf merged 3 commits intor0.5.0from
rywolf/nsfw-model
Oct 16, 2024
Merged

Change NSFW Model#307
ryantwolf merged 3 commits intor0.5.0from
rywolf/nsfw-model

Conversation

@ryantwolf
Copy link
Contributor

@ryantwolf ryantwolf commented Oct 16, 2024

Description

Changes the NSFW model from one that takes ViT-H embeddings as input into one that takes ViT-L embeddings as input.

Usage

No change

Checklist

  • I am familiar with the Contributing Guide.
  • New or Existing tests cover these changes.
  • The documentation is up to date with these changes.

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>
Signed-off-by: Ryan Wolf <rywolf@nvidia.com>
Signed-off-by: Ryan Wolf <rywolf@nvidia.com>
@ryantwolf ryantwolf requested a review from VibhuJawa October 16, 2024 21:14
Copy link
Contributor

@VibhuJawa VibhuJawa left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks for fixing this

@ryantwolf ryantwolf merged commit 37058a9 into r0.5.0 Oct 16, 2024
@ryantwolf ryantwolf deleted the rywolf/nsfw-model branch October 16, 2024 22:41
ayushdg pushed a commit to ayushdg/NeMo-Curator that referenced this pull request Oct 30, 2024
* Change download for NSFW model

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Fix model init

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Fix embedding size

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

---------

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>
Signed-off-by: Ayush Dattagupta <ayushdg95@gmail.com>
ayushdg added a commit that referenced this pull request Nov 4, 2024
* Bump version

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Pin Rapids to 24.8 for 0.5.0 Release  (#273)

* pin to 24.8

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* remove nemo_toolkit

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* restrict spacy

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* unrestrict spacy

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* restrict spacy + numpy

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

---------

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* Fix tutorial for release (#277)

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Fix nworkers for tinystories tutorial (#280)

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Pin to 24.8.x instead of 24.8 (#282)

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* Change NSFW Model (#307)

* Change download for NSFW model

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Fix model init

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Fix embedding size

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

---------

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Bump version to 0.6.0.dev0

Signed-off-by: Ayush Dattagupta <ayushdg95@gmail.com>

* added error handling (#343)

Signed-off-by: Tanay Varshney <tvarshney@nvidia.com>

---------

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>
Signed-off-by: Praateek Mahajan <praateekm@gmail.com>
Signed-off-by: Ayush Dattagupta <ayushdg95@gmail.com>
Signed-off-by: Tanay Varshney <tvarshney@nvidia.com>
Co-authored-by: Ryan Wolf <rywolf@nvidia.com>
Co-authored-by: Praateek Mahajan <praateekmahajan@users.noreply.github.com>
Co-authored-by: Tanay Varshney <varshneytanay@gmail.com>
vinay-raman pushed a commit to vinay-raman/NeMo-Curator that referenced this pull request Nov 26, 2024
* Bump version

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Pin Rapids to 24.8 for 0.5.0 Release  (NVIDIA-NeMo#273)

* pin to 24.8

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* remove nemo_toolkit

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* restrict spacy

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* unrestrict spacy

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* restrict spacy + numpy

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

---------

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* Fix tutorial for release (NVIDIA-NeMo#277)

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Fix nworkers for tinystories tutorial (NVIDIA-NeMo#280)

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Pin to 24.8.x instead of 24.8 (NVIDIA-NeMo#282)

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* Change NSFW Model (NVIDIA-NeMo#307)

* Change download for NSFW model

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Fix model init

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Fix embedding size

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

---------

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Bump version to 0.6.0.dev0

Signed-off-by: Ayush Dattagupta <ayushdg95@gmail.com>

* added error handling (NVIDIA-NeMo#343)

Signed-off-by: Tanay Varshney <tvarshney@nvidia.com>

---------

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>
Signed-off-by: Praateek Mahajan <praateekm@gmail.com>
Signed-off-by: Ayush Dattagupta <ayushdg95@gmail.com>
Signed-off-by: Tanay Varshney <tvarshney@nvidia.com>
Co-authored-by: Ryan Wolf <rywolf@nvidia.com>
Co-authored-by: Praateek Mahajan <praateekmahajan@users.noreply.github.com>
Co-authored-by: Tanay Varshney <varshneytanay@gmail.com>
Signed-off-by: Vinay Raman <viraman@nvidia.com>
jnke2016 pushed a commit to jnke2016/Curator that referenced this pull request Nov 12, 2025
* Bump version

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Pin Rapids to 24.8 for 0.5.0 Release  (NVIDIA-NeMo#273)

* pin to 24.8

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* remove nemo_toolkit

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* restrict spacy

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* unrestrict spacy

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* restrict spacy + numpy

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

---------

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* Fix tutorial for release (NVIDIA-NeMo#277)

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Fix nworkers for tinystories tutorial (NVIDIA-NeMo#280)

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Pin to 24.8.x instead of 24.8 (NVIDIA-NeMo#282)

Signed-off-by: Praateek Mahajan <praateekm@gmail.com>

* Change NSFW Model (NVIDIA-NeMo#307)

* Change download for NSFW model

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Fix model init

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Fix embedding size

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

---------

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>

* Bump version to 0.6.0.dev0

Signed-off-by: Ayush Dattagupta <ayushdg95@gmail.com>

* added error handling (NVIDIA-NeMo#343)

Signed-off-by: Tanay Varshney <tvarshney@nvidia.com>

---------

Signed-off-by: Ryan Wolf <rywolf@nvidia.com>
Signed-off-by: Praateek Mahajan <praateekm@gmail.com>
Signed-off-by: Ayush Dattagupta <ayushdg95@gmail.com>
Signed-off-by: Tanay Varshney <tvarshney@nvidia.com>
Co-authored-by: Ryan Wolf <rywolf@nvidia.com>
Co-authored-by: Praateek Mahajan <praateekmahajan@users.noreply.github.com>
Co-authored-by: Tanay Varshney <varshneytanay@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants