TaahKay commited on
Commit
bc5d040
·
verified ·
1 Parent(s): 44859bd

Update models/model_loader.py

Browse files
Files changed (1) hide show
  1. 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(10) # give system time to clean up before moving on
 
42
  del model
43
  torch.cuda.empty_cache()
44
  gc.collect()
45
+ time.sleep(5) # give system time to clean up before moving on