mrm8488 commited on
Commit
ade9726
1 Parent(s): ff94fa2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -63,6 +63,8 @@ pip install bitsandbytes-cuda111==0.26.0
63
  import transformers
64
  import torch
65
 
 
 
66
 
67
  class GPTJBlock(transformers.models.gptj.modeling_gptj.GPTJBlock):
68
  def __init__(self, config):
 
63
  import transformers
64
  import torch
65
 
66
+ from utils import convert_to_int8
67
+
68
 
69
  class GPTJBlock(transformers.models.gptj.modeling_gptj.GPTJBlock):
70
  def __init__(self, config):