Bug Fix: Check for explicit cli device (fast)

[Jack-Khuu]

This fixes a bug where a user provided "fast" device arg falls through without being converted to a device. This correctly catches this case.

Bug introduced in 93f713f,

---

Command

python3 torchchat.py generate llama3.1 --prompt "write me a story about a boy and his bear" --device fast

Original Failure:

RuntimeError: don't know how to restore data location of torch.storage.UntypedStorage (tagged with fast)

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/torchchat/1374

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

• GLIBC not found in Nova workflows

✅ No Failures

As of commit 703786c with merge base 2cf1a17 ():
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.