Specifically import torch.cuda.comm in breakmodel.py

KoboldAI · Aug 20, 2021 · eef0db8 · eef0db8
1 parent b1c13f8
commit eef0db8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/breakmodel.py b/breakmodel.py
@@ -212,6 +212,7 @@
 
 
 import torch
+import torch.cuda.comm
 import copy
 import gc