bwilkie commited on
Commit
36affcb
·
verified ·
1 Parent(s): 0006d3c

Update myagent.py

Browse files
Files changed (1) hide show
  1. myagent.py +1 -2
myagent.py CHANGED
@@ -12,8 +12,7 @@ import torch
12
  # Basic model wrapper for local inference with debug info
13
  class BasicAgent:
14
  def __init__(self, model, tokenizer):
15
- self.model = model
16
- self.tokenizer = tokenizer
17
  self.device = model.device if hasattr(model, 'device') else 'cpu'
18
  print(f"Model device: {self.device}")
19
 
 
12
  # Basic model wrapper for local inference with debug info
13
  class BasicAgent:
14
  def __init__(self, model, tokenizer):
15
+
 
16
  self.device = model.device if hasattr(model, 'device') else 'cpu'
17
  print(f"Model device: {self.device}")
18