ubiodee commited on
Commit
d32b4dc
·
verified ·
1 Parent(s): 8f2fe34

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +9 -2
config.json CHANGED
@@ -1,5 +1,12 @@
1
  {
2
- "api": {
3
- "task": "text-generation"
 
 
 
 
 
 
4
  }
 
5
  }
 
1
  {
2
+ "architectures": [
3
+ "LlamaForCausalLM"
4
+ ],
5
+ "model_type": "llama",
6
+ "task_specific_params": {
7
+ "text-generation": {
8
+ "max_length": 200,
9
+ "do_sample": true
10
  }
11
+ }
12
  }