Spaces:
Sleeping
Sleeping
henok3878
commited on
Commit
·
91f531c
1
Parent(s):
26ec32c
try the full model (no quantization)
Browse files
main.py
CHANGED
|
@@ -63,8 +63,8 @@ async def lifespan(app: FastAPI):
|
|
| 63 |
|
| 64 |
scripted_model_path = MODEL_DIR / SCRIPTED_MODEL_NAME
|
| 65 |
metadata_model_path = MODEL_DIR / METADATA_MODEL_NAME
|
| 66 |
-
if DEVICE.type == "cpu":
|
| 67 |
-
|
| 68 |
|
| 69 |
if not scripted_model_path.exists():
|
| 70 |
logger.error(f"Traced model not found at {scripted_model_path}")
|
|
|
|
| 63 |
|
| 64 |
scripted_model_path = MODEL_DIR / SCRIPTED_MODEL_NAME
|
| 65 |
metadata_model_path = MODEL_DIR / METADATA_MODEL_NAME
|
| 66 |
+
# if DEVICE.type == "cpu":
|
| 67 |
+
# scripted_model_path = MODEL_DIR / QUANTIZED_MODEL_NAME
|
| 68 |
|
| 69 |
if not scripted_model_path.exists():
|
| 70 |
logger.error(f"Traced model not found at {scripted_model_path}")
|