Spaces:
Sleeping
Sleeping
Commit
·
f6dcb7a
1
Parent(s):
13373dd
fix
Browse files- config_store.py +15 -10
- pre-requirements.txt +6 -0
config_store.py
CHANGED
@@ -78,24 +78,29 @@ def get_pytorch_config():
|
|
78 |
|
79 |
def get_openvino_config():
|
80 |
return {
|
81 |
-
"openvino.use_cache": gr.Checkbox(
|
82 |
-
value=True,
|
83 |
-
label="openvino.use_cache",
|
84 |
-
info="Uses the decoder with cache if available",
|
85 |
-
),
|
86 |
-
"openvino.use_merged": gr.Checkbox(
|
87 |
-
value=True,
|
88 |
-
label="openvino.use_merged",
|
89 |
-
info="Uses merged model if available",
|
90 |
-
),
|
91 |
"openvino.reshape": gr.Checkbox(
|
92 |
value=False,
|
93 |
label="openvino.reshape",
|
94 |
info="Reshapes the model to the input shape",
|
95 |
),
|
|
|
|
|
|
|
|
|
|
|
96 |
"openvino.half": gr.Checkbox(
|
97 |
value=False,
|
98 |
label="openvino.half",
|
99 |
info="Converts model to half precision",
|
100 |
),
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
101 |
}
|
|
|
78 |
|
79 |
def get_openvino_config():
|
80 |
return {
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
81 |
"openvino.reshape": gr.Checkbox(
|
82 |
value=False,
|
83 |
label="openvino.reshape",
|
84 |
info="Reshapes the model to the input shape",
|
85 |
),
|
86 |
+
"openvino.compile": gr.Checkbox(
|
87 |
+
value=False,
|
88 |
+
label="openvino.compile",
|
89 |
+
info="Compiles model for the current device",
|
90 |
+
),
|
91 |
"openvino.half": gr.Checkbox(
|
92 |
value=False,
|
93 |
label="openvino.half",
|
94 |
info="Converts model to half precision",
|
95 |
),
|
96 |
+
"openvino.load_in_8bit": gr.Checkbox(
|
97 |
+
value=False,
|
98 |
+
label="openvino.load_in_8bit",
|
99 |
+
info="Loads model in 8 bits precision",
|
100 |
+
),
|
101 |
+
"openvino.load_in_4bit": gr.Checkbox(
|
102 |
+
value=False,
|
103 |
+
label="openvino.load_in_4bit",
|
104 |
+
info="Loads model in 4 bits precision",
|
105 |
+
),
|
106 |
}
|
pre-requirements.txt
CHANGED
@@ -1 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
pip==24.3.1
|
|
|
1 |
+
--index-url https://download.pytorch.org/whl/cpu
|
2 |
+
torch
|
3 |
+
--index-url https://download.pytorch.org/whl/cpu
|
4 |
+
torchvision
|
5 |
+
--index-url https://download.pytorch.org/whl/cpu
|
6 |
+
torchaudio
|
7 |
pip==24.3.1
|