Spaces:
Sleeping
Sleeping
jordigonzm
commited on
Commit
•
25dc2bf
1
Parent(s):
3b365d6
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import gradio as gr
|
|
6 |
|
7 |
# Constantes de configuración
|
8 |
MODEL_NAME = 'google/flan-t5-base'
|
9 |
-
MODEL_PATH = 'models/flan-t5-base-
|
10 |
QUANTIZE_DTYPE = torch.qint8
|
11 |
QUANTIZE_MODULES = {torch.nn.Linear}
|
12 |
MAX_INPUT_LENGTH = 512
|
|
|
6 |
|
7 |
# Constantes de configuración
|
8 |
MODEL_NAME = 'google/flan-t5-base'
|
9 |
+
MODEL_PATH = 'models/flan-t5-base-quantized.pth'
|
10 |
QUANTIZE_DTYPE = torch.qint8
|
11 |
QUANTIZE_MODULES = {torch.nn.Linear}
|
12 |
MAX_INPUT_LENGTH = 512
|