Update myagent.py
Browse files- 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 |
-
|
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 |
|