chinoll
/

chatsakura-3b-int4

Text Generation

text-generation-inference

Model card Files Files and versions Community

chinoll commited on Apr 1, 2023

Commit

fabe522

·

1 Parent(s): cf462e1

Update modeling_chatsakura.py

Files changed (1) hide show

modeling_chatsakura.py +2 -2

modeling_chatsakura.py CHANGED Viewed

@@ -1,9 +1,9 @@
 import torch
 import torch.nn as nn
-from gptq import *
 # from modelutils import *
-from quant import *
 from transformers import BloomForCausalLM as LM
 def find_layers(module, layers=[nn.Conv2d, nn.Linear], name=''):

 import torch
 import torch.nn as nn
+from .gptq import *
 # from modelutils import *
+from .quant import *
 from transformers import BloomForCausalLM as LM
 def find_layers(module, layers=[nn.Conv2d, nn.Linear], name=''):