jadechoghari commited on
Commit
3a6f4ec
·
verified ·
1 Parent(s): 19fb693

add automodel

Browse files
Files changed (1) hide show
  1. config.json +5 -1
config.json CHANGED
@@ -15,5 +15,9 @@
15
  "transformers_version": "4.42.4",
16
  "triplane_dim": 80,
17
  "triplane_high_res": 64,
18
- "triplane_low_res": 32
 
 
 
 
19
  }
 
15
  "transformers_version": "4.42.4",
16
  "triplane_dim": 80,
17
  "triplane_high_res": 64,
18
+ "triplane_low_res": 32,
19
+ "auto_map": {
20
+ "AutoConfig": "modeling.LRMGeneratorConfig",
21
+ "AutoModel": "modeling.LRMGenerator"
22
+ }
23
  }