ASethi04 commited on
Commit
298a67b
·
verified ·
1 Parent(s): a86f0c2

Initial upload of VINE model architecture and weights - config

Browse files
Files changed (1) hide show
  1. config.json +3 -2
config.json CHANGED
@@ -1,5 +1,6 @@
1
  {
2
- "_device": "cpu",
 
3
  "alpha": 0.5,
4
  "auto_map": {
5
  "AutoConfig": "vine_config.VineConfig"
@@ -25,7 +26,7 @@
25
  "target_fps": 1,
26
  "text_threshold": 0.25,
27
  "topk_cate": 3,
28
- "transformers_version": "4.57.2",
29
  "use_hf_repo": true,
30
  "visualization_dir": null,
31
  "visualize": false,
 
1
  {
2
+ "_attn_implementation_autoset": true,
3
+ "_device": "cuda",
4
  "alpha": 0.5,
5
  "auto_map": {
6
  "AutoConfig": "vine_config.VineConfig"
 
26
  "target_fps": 1,
27
  "text_threshold": 0.25,
28
  "topk_cate": 3,
29
+ "transformers_version": "4.46.2",
30
  "use_hf_repo": true,
31
  "visualization_dir": null,
32
  "visualize": false,