Merge pull request #238 from YellowRoseCx/patch-1

adding missing transformers import to opt.py old-cuda
qwopqwop200 · May 18, 2023 · 8e78736 · 8e78736
2 parents b5f9188 + 4cf6cf6
commit 8e78736
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/opt.py b/opt.py
@@ -2,6 +2,7 @@
 
 import torch
 import torch.nn as nn
+import transformers
 
 from gptq import *
 from modelutils import *