Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Support torch.bernoulli operator. #170

Closed
Tracked by #179
tfogal opened this issue Apr 12, 2024 · 0 comments 路 Fixed by #297
Closed
Tracked by #179

Support torch.bernoulli operator. #170

tfogal opened this issue Apr 12, 2024 · 0 comments 路 Fixed by #297
Assignees
@kshitij12345
Labels
enhancement New feature or request MegatronImagen Needed to support NeMo's MegatronImagen model (text to image generation) nemo Issues needed to support NVIDIA NeMo models. operators

Comments

@tfogal
Copy link
Collaborator

tfogal commented Apr 12, 2024 •
edited by github-actions bot
Loading

馃殌 Feature

Please support torch.bernoulli in Thunder.

Motivation

Trying to support NeMo's multimodal text-to-image model.

cc @apaz-cli @tfogal
@tfogal tfogal added enhancement New feature or request nemo Issues needed to support NVIDIA NeMo models. MegatronImagen Needed to support NeMo's MegatronImagen model (text to image generation) labels Apr 12, 2024
@tfogal tfogal mentioned this issue Apr 12, 2024
Open
13 tasks
@kshitij12345 kshitij12345 self-assigned this Apr 29, 2024
@kshitij12345 kshitij12345 mentioned this issue Apr 29, 2024
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@kshitij12345 kshitij12345

Labels
enhancement New feature or request MegatronImagen Needed to support NeMo's MegatronImagen model (text to image generation) nemo Issues needed to support NVIDIA NeMo models. operators
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

add torch.bernoulli kshitij12345/lightning-thunder
3 participants
@tfogal @riccardofelluga @kshitij12345