Update t2tmod.py

t2tmod.py

@@ -33,7 +33,7 @@ class TaggerLlama:
             return
         print("Downloading Model")
         self.tokenizer = AutoTokenizer.from_pretrained(self.MODEL_URL)
-        self.model = AutoModelForCausalLM.from_pretrained(self.MODEL_URL, device_map="cuda:0")
+        self.model = AutoModelForCausalLM.from_pretrained(self.MODEL_URL, device_map="cpu")
         self.tokenizer.save_pretrained(self.SAVE_NAME)
         self.model.save_pretrained(self.SAVE_NAME)
         print("Model Downloaded")