Update services/base_generator.py
Browse files
services/base_generator.py
CHANGED
|
@@ -3,6 +3,7 @@ from abc import ABC, abstractmethod
|
|
| 3 |
from typing import AsyncGenerator, Dict, Any, Optional, List, Tuple
|
| 4 |
from dataclasses import dataclass
|
| 5 |
from logging import getLogger
|
|
|
|
| 6 |
|
| 7 |
|
| 8 |
from config.config import GenerationConfig, ModelConfig
|
|
|
|
| 3 |
from typing import AsyncGenerator, Dict, Any, Optional, List, Tuple
|
| 4 |
from dataclasses import dataclass
|
| 5 |
from logging import getLogger
|
| 6 |
+
from services.model_manager import ModelManager
|
| 7 |
|
| 8 |
|
| 9 |
from config.config import GenerationConfig, ModelConfig
|