Fix import and typo in device

by rk404 - opened Mar 11, 2024

←

rk404

Mar 11, 2024

pipeline wasn't imported so it was giving error.
It should be "cpu" instead of "CPU" in device=torch.device("cuda" if torch.cuda.is_available() else "CPU"),

Protect AI org Mar 11, 2024

Thank you, @rk404 .

asofter changed pull request status to merged Mar 11, 2024

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

Tap or paste here to upload images

Your need to confirm your account before you can post a new comment.

· Sign up or log in to comment