Use GPU when possible

by AniketGupta895 - opened 17 days ago

←

17 days ago

Model was hardcoded to use the cpu. This change will move to GPU using .to(device) instead of .to("cpu")

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Ready to merge

This branch is ready to get merged automatically.

· Sign up or log in to comment