Hugging Face's logo

Spaces:
p1atdev
/
llm-jp-3-instruct2-demo
Sleeping

App Files Community
p1atdev commited on
Commit
5065181
·
verified ·
1 Parent(s): 456350b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -30,13 +30,13 @@ import spaces
30
 
31
 
32
  MODEL_NAME_MAP = {
33
- "150M": "llm-jp/llm-jp-2-150m-instruct3",
34
- "440M": "llm-jp/llm-jp-2-440m-instruct3",
35
- "980M": "llm-jp/llm-jp-2-980m-instruct3",
36
- "1.8B": "llm-jp/llm-jp-2-1.8b-instruct3",
37
- "3.7B": "llm-jp/llm-jp-2-3.7b-instruct3",
38
- "7.2B": "llm-jp/llm-jp-2-7.2b-instruct3",
39
- "13B": "llm-jp/llm-jp-2-13b-instruct3",
40
  }
41
 
42
  quantization_config = BitsAndBytesConfig(
 
30
 
31
 
32
  MODEL_NAME_MAP = {
33
+ "150M": "llm-jp/llm-jp-3-150m-instruct2",
34
+ "440M": "llm-jp/llm-jp-3-440m-instruct2",
35
+ "980M": "llm-jp/llm-jp-3-980m-instruct2",
36
+ "1.8B": "llm-jp/llm-jp-3-1.8b-instruct2",
37
+ "3.7B": "llm-jp/llm-jp-3-3.7b-instruct2",
38
+ "7.2B": "llm-jp/llm-jp-3-7.2b-instruct2",
39
+ "13B": "llm-jp/llm-jp-3-13b-instruct2",
40
  }
41
 
42
  quantization_config = BitsAndBytesConfig(