Update backend/worker.py
Browse files- backend/worker.py +1 -1
backend/worker.py
CHANGED
|
@@ -6,7 +6,7 @@ import traceback
|
|
| 6 |
from merge_engines.linear import merge_models
|
| 7 |
from merge_engines.evolutionary import evolutionary_merge
|
| 8 |
from uploader import upload_to_hf
|
| 9 |
-
from memory_manager import
|
| 10 |
import asyncio
|
| 11 |
|
| 12 |
async def run_job_async(job_id: str, params: dict):
|
|
|
|
| 6 |
from merge_engines.linear import merge_models
|
| 7 |
from merge_engines.evolutionary import evolutionary_merge
|
| 8 |
from uploader import upload_to_hf
|
| 9 |
+
from memory_manager import check_memory_safe
|
| 10 |
import asyncio
|
| 11 |
|
| 12 |
async def run_job_async(job_id: str, params: dict):
|