chore: Import torch module in model.py
Browse files- kitt/core/model.py +1 -0
kitt/core/model.py
CHANGED
|
@@ -13,6 +13,7 @@ from langchain_core.utils.function_calling import convert_to_openai_tool
|
|
| 13 |
from loguru import logger
|
| 14 |
from ollama import Client
|
| 15 |
from pydantic import BaseModel
|
|
|
|
| 16 |
|
| 17 |
from kitt.skills import vehicle_status
|
| 18 |
from kitt.skills.common import config
|
|
|
|
| 13 |
from loguru import logger
|
| 14 |
from ollama import Client
|
| 15 |
from pydantic import BaseModel
|
| 16 |
+
import torch
|
| 17 |
|
| 18 |
from kitt.skills import vehicle_status
|
| 19 |
from kitt.skills.common import config
|