A.I.StateMachine

Build error

Chris4K commited on Jan 27, 2025

Commit

3d8d158

verified ·

1 Parent(s): e1499dd

Update services/base_generator.py

Files changed (1) hide show

services/base_generator.py CHANGED Viewed

@@ -3,6 +3,7 @@ from abc import ABC, abstractmethod
 from typing import AsyncGenerator, Dict, Any, Optional, List, Tuple
 from dataclasses import dataclass
 from logging import getLogger
 from config.config import GenerationConfig, ModelConfig

 from typing import AsyncGenerator, Dict, Any, Optional, List, Tuple
 from dataclasses import dataclass
 from logging import getLogger
+from services.model_manager import ModelManager
 from config.config import GenerationConfig, ModelConfig