Spaces:
Sleeping
Sleeping
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 |
|