Cyleux
/

danielclone2

Model card Files Files and versions

Cyleux commited on May 17, 2023

Commit

908ce46

·

1 Parent(s): a901e5b

Update chat_final.py

Files changed (1) hide show

chat_final.py +1 -1

chat_final.py CHANGED Viewed

@@ -15,7 +15,7 @@ UNKNOWN_CHAR = None
 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'

 tokenizer = TOKENIZER(WORD_NAME, UNKNOWN_CHAR=UNKNOWN_CHAR)
 args = types.SimpleNamespace()
+args.RUN_DEVICE = "cuda"
 args.FLOAT_MODE = "fp32"
 args.vocab_size = 50277
 args.MODEL_NAME = 'zrwkv-37fifth'