Cyleux
/

danielclone2

Model card Files Files and versions Community

Cyleux commited on May 17, 2023

Commit

a901e5b

·

1 Parent(s): 672cd49

Update chat_final.py

Files changed (1) hide show

chat_final.py +3 -3

chat_final.py CHANGED Viewed

@@ -15,11 +15,11 @@ UNKNOWN_CHAR = None
 tokenizer = TOKENIZER(WORD_NAME, UNKNOWN_CHAR=UNKNOWN_CHAR)
 args = types.SimpleNamespace()
-args.RUN_DEVICE = "cpu"
 args.FLOAT_MODE = "fp32"
 args.vocab_size = 50277
-# args.MODEL_NAME = 'zrwkv-37fifth'
-args.MODEL_NAME = 'zrwkv-23fifth'
 args.n_layer = 12
 args.n_embd = 768

 tokenizer = TOKENIZER(WORD_NAME, UNKNOWN_CHAR=UNKNOWN_CHAR)
 args = types.SimpleNamespace()
+args.RUN_DEVICE = "gpu"
 args.FLOAT_MODE = "fp32"
 args.vocab_size = 50277
+args.MODEL_NAME = 'zrwkv-37fifth'
+# args.MODEL_NAME = 'zrwkv-23fifth'
 args.n_layer = 12
 args.n_embd = 768