mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-04-21 14:59:02 +00:00
git status
This commit is contained in:
1
toolkit/memory_management/__init__.py
Normal file
1
toolkit/memory_management/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
from .manager import MemoryManager
|
||||
12
toolkit/memory_management/manager.py
Normal file
12
toolkit/memory_management/manager.py
Normal file
@@ -0,0 +1,12 @@
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from toolkit.models.base_model import BaseModel
|
||||
|
||||
|
||||
class MemoryManager:
|
||||
def __init__(
|
||||
self,
|
||||
model: "BaseModel",
|
||||
):
|
||||
self.model: "BaseModel" = model
|
||||
Reference in New Issue
Block a user