Spaces:
Runtime error
Runtime error
Commit
·
7659f48
1
Parent(s):
204e8d8
Update models/llama.py
Browse files- models/llama.py +1 -0
models/llama.py
CHANGED
|
@@ -21,6 +21,7 @@ class BnbInt8Linear(bnb.nn.Linear8bitLt):
|
|
| 21 |
|
| 22 |
|
| 23 |
def get_linear_layer(use_int8):
|
|
|
|
| 24 |
if use_int8:
|
| 25 |
return BnbInt8Linear
|
| 26 |
return NormalLinear
|
|
|
|
| 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
|