rikunarita-2 commited on
Commit
657f21b
·
verified ·
1 Parent(s): 8af2a15

Update backend/worker.py

Browse files
Files changed (1) hide show
  1. 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 estimate_memory, check_memory_safe
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):