Spaces:
Running
Running
Benjamin Consolvo
commited on
Commit
•
6dba019
1
Parent(s):
6128b93
neural_compressor
Browse files- app.py +1 -1
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
from transformers import pipeline
|
| 3 |
import time
|
| 4 |
-
|
| 5 |
from optimum.intel.neural_compressor import IncQuantizedModelForQuestionAnswering
|
| 6 |
|
| 7 |
# model_id = "Intel/bert-base-uncased-squadv1.1-sparse-80-1x4-block-pruneofa"
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
from transformers import pipeline
|
| 3 |
import time
|
| 4 |
+
import neural_compressor
|
| 5 |
from optimum.intel.neural_compressor import IncQuantizedModelForQuestionAnswering
|
| 6 |
|
| 7 |
# model_id = "Intel/bert-base-uncased-squadv1.1-sparse-80-1x4-block-pruneofa"
|
requirements.txt
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
transformers
|
| 2 |
torch
|
| 3 |
tensorflow
|
|
|
|
| 4 |
optimum.intel
|
|
|
|
| 1 |
transformers
|
| 2 |
torch
|
| 3 |
tensorflow
|
| 4 |
+
neural_compressor
|
| 5 |
optimum.intel
|