Update models/model_loader.py
Browse files- models/model_loader.py +1 -1
models/model_loader.py
CHANGED
@@ -42,4 +42,4 @@ def unload_model(model):
|
|
42 |
del model
|
43 |
torch.cuda.empty_cache()
|
44 |
gc.collect()
|
45 |
-
time.sleep(
|
|
|
42 |
del model
|
43 |
torch.cuda.empty_cache()
|
44 |
gc.collect()
|
45 |
+
time.sleep(5) # give system time to clean up before moving on
|