Update 3 files

- /dataset.py
- /model.py
- /trainer.cli.py

dataset.py

@@ -31,7 +31,7 @@ class Dataset:
 
         batches = []
         for batch in array_reshaped:
-            tensor_batch = torch.tensor(batch, dtype=torch.long)#.to(GetDevice())
+            tensor_batch = torch.tensor(batch, dtype=torch.long).to(GetDevice())
             batches.append(tensor_batch)
 
         return batches, num_batches

model.py

@@ -50,12 +50,12 @@ class Model:
 
         with torch.no_grad():
             encoded_ids = tokenizer.encode(seed_text)
-            input_ids = torch.tensor(encoded_ids).unsqueeze(0)#.to(GetDevice())
+            input_ids = torch.tensor(encoded_ids).unsqueeze(0).to(GetDevice())
             output = self.model.generate(input_ids, max_length=max_len)
 
             logits = output[0].tolist()
             text = tokenizer.decode(logits)
-
+            
         return text
 
 

trainer.cli.py

@@ -52,4 +52,9 @@ if __name__ == '__main__':
 
 
     trainer = Trainer(config.trainer)
-    trainer.train(batches)
+    
+    while True:
+        pass 
+
+        
+    #trainer.train(batches)