Hugging Face's logo

Cyleux
/
danielclone2

Model card Files
xet
Community
Cyleux commited on
Commit
a901e5b
·
1 Parent(s): 672cd49

Update chat_final.py

Browse files
Files changed (1) hide show
  1. chat_final.py +3 -3
chat_final.py CHANGED
@@ -15,11 +15,11 @@ UNKNOWN_CHAR = None
15
  tokenizer = TOKENIZER(WORD_NAME, UNKNOWN_CHAR=UNKNOWN_CHAR)
16
 
17
  args = types.SimpleNamespace()
18
- args.RUN_DEVICE = "cpu"
19
  args.FLOAT_MODE = "fp32"
20
  args.vocab_size = 50277
21
- # args.MODEL_NAME = 'zrwkv-37fifth'
22
- args.MODEL_NAME = 'zrwkv-23fifth'
23
 
24
  args.n_layer = 12
25
  args.n_embd = 768
 
15
  tokenizer = TOKENIZER(WORD_NAME, UNKNOWN_CHAR=UNKNOWN_CHAR)
16
 
17
  args = types.SimpleNamespace()
18
+ args.RUN_DEVICE = "gpu"
19
  args.FLOAT_MODE = "fp32"
20
  args.vocab_size = 50277
21
+ args.MODEL_NAME = 'zrwkv-37fifth'
22
+ # args.MODEL_NAME = 'zrwkv-23fifth'
23
 
24
  args.n_layer = 12
25
  args.n_embd = 768