IlyasMoutawwakil HF staff commited on
Commit
f6dcb7a
·
1 Parent(s): 13373dd
Files changed (2) hide show
  1. config_store.py +15 -10
  2. 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