henok3878 commited on
Commit
91f531c
·
1 Parent(s): 26ec32c

try the full model (no quantization)

Browse files
Files changed (1) hide show
  1. main.py +2 -2
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
- 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}")
 
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}")