Spaces:
Running
Running
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}")
|