Spaces:
Runtime error
Runtime error
Commit
·
f61a558
1
Parent(s):
548befc
Update generate.py
Browse files- generate.py +1 -1
generate.py
CHANGED
|
@@ -21,7 +21,7 @@ TYPE_WRITER=1 # whether output streamly
|
|
| 21 |
|
| 22 |
args = parser.parse_args()
|
| 23 |
print(args)
|
| 24 |
-
tokenizer = LlamaTokenizer.from_pretrained(
|
| 25 |
|
| 26 |
LOAD_8BIT = True
|
| 27 |
|
|
|
|
| 21 |
|
| 22 |
args = parser.parse_args()
|
| 23 |
print(args)
|
| 24 |
+
tokenizer = LlamaTokenizer.from_pretrained(BASE_MODEL)
|
| 25 |
|
| 26 |
LOAD_8BIT = True
|
| 27 |
|