laurenssam commited on
Commit
afaf26e
·
1 Parent(s): 54ee376

test with float32

Browse files
Files changed (1) hide show
  1. tinyllava_demo/config.json +1 -1
tinyllava_demo/config.json CHANGED
@@ -42,7 +42,7 @@
42
  "tie_word_embeddings": false,
43
  "tokenizer_model_max_length": 3072,
44
  "tokenizer_padding_side": "right",
45
- "torch_dtype": "float32",
46
  "transformers_version": "4.37.2",
47
  "tune_embed_tokens": false,
48
  "tune_entire_model": false,
 
42
  "tie_word_embeddings": false,
43
  "tokenizer_model_max_length": 3072,
44
  "tokenizer_padding_side": "right",
45
+ "torch_dtype": "float16",
46
  "transformers_version": "4.37.2",
47
  "tune_embed_tokens": false,
48
  "tune_entire_model": false,