Fix import and typo in device
#1
by
rk404
- opened
- 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"),
asofter
changed pull request status to
merged