From b6e3910fd0f6835ef759e6bf3a96f50f58a19da2 Mon Sep 17 00:00:00 2001
From: Nripesh Niketan <86844847+NripeshN@users.noreply.github.com>
Date: Thu, 30 Jan 2025 16:04:00 +0000
Subject: [PATCH] Fix small error

---
 inference/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inference/model.py b/inference/model.py
index 3d8bf2c..18586da 100644
--- a/inference/model.py
+++ b/inference/model.py
@@ -802,7 +802,7 @@ if __name__ == "__main__":
         default_device = "mps"
     else:
         default_device = "cpu"
-    torch.set_default_device("default_device")
+    torch.set_default_device(default_device)
     torch.manual_seed(0)
     args = ModelArgs()
     x = torch.randint(0, args.vocab_size, (2, 128))