Upload mod.py
Browse files
mod.py
CHANGED
|
@@ -76,7 +76,7 @@ def is_repo_exists(repo_id):
|
|
| 76 |
def clear_cache():
|
| 77 |
try:
|
| 78 |
torch.cuda.empty_cache()
|
| 79 |
-
torch.cuda.reset_max_memory_allocated()
|
| 80 |
torch.cuda.reset_peak_memory_stats()
|
| 81 |
gc.collect()
|
| 82 |
except Exception as e:
|
|
|
|
| 76 |
def clear_cache():
|
| 77 |
try:
|
| 78 |
torch.cuda.empty_cache()
|
| 79 |
+
#torch.cuda.reset_max_memory_allocated()
|
| 80 |
torch.cuda.reset_peak_memory_stats()
|
| 81 |
gc.collect()
|
| 82 |
except Exception as e:
|