Spaces:
Sleeping
Sleeping
prasanna kumar
commited on
Commit
•
43be51d
1
Parent(s):
eb89467
added gemma model support
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ model_path = "models/"
|
|
6 |
import gradio as gr
|
7 |
|
8 |
# Available models
|
9 |
-
MODELS = ["Meta-Llama-3.1-8B"]
|
10 |
|
11 |
def process_input(input_type, input_value, model_name):
|
12 |
# Initialize tokenizer
|
|
|
6 |
import gradio as gr
|
7 |
|
8 |
# Available models
|
9 |
+
MODELS = ["Meta-Llama-3.1-8B","gemma-2b"]
|
10 |
|
11 |
def process_input(input_type, input_value, model_name):
|
12 |
# Initialize tokenizer
|
models/gemma-2b/model.safetensors.index.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1b32948684e244670c6d7c9a9dc723bce5d4b2b34319879c9c1844a0cbf1a15a
|
3 |
+
size 13489
|
models/gemma-2b/special_tokens_map.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:baec30ea10906f16adb8c18af7a34023002c1746542612b8b41c9f09e1351351
|
3 |
+
size 636
|
models/gemma-2b/tokenizer.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c15eb04bc5ad609fb26533e8525302c5640a945e5f67f65b7c849900acda7d99
|
3 |
+
size 17518497
|
models/gemma-2b/tokenizer.model
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:61a7b147390c64585d6c3543dd6fc636906c9af3865a5548f27f31aee1d4c8e2
|
3 |
+
size 4241003
|
models/gemma-2b/tokenizer_config.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d1cc6e515e5900205fcc07b62fd16503bbff92b9c2c22bed56fd64400402fbb5
|
3 |
+
size 33556
|