Spaces:
Runtime error
Runtime error
yuhaofeng-shiba
commited on
Commit
•
6259eea
1
Parent(s):
784d667
Update models/llama.py
Browse files- models/llama.py +0 -1
models/llama.py
CHANGED
@@ -21,7 +21,6 @@ class BnbInt8Linear(bnb.nn.Linear8bitLt):
|
|
21 |
|
22 |
|
23 |
def get_linear_layer(use_int8):
|
24 |
-
print('use_int8:', use_int8)
|
25 |
if use_int8:
|
26 |
return BnbInt8Linear
|
27 |
return NormalLinear
|
|
|
21 |
|
22 |
|
23 |
def get_linear_layer(use_int8):
|
|
|
24 |
if use_int8:
|
25 |
return BnbInt8Linear
|
26 |
return NormalLinear
|