andstor commited on
Commit
6640473
1 Parent(s): 7fe2695

Update src/app.py

Browse files
Files changed (1) hide show
  1. src/app.py +1 -1
src/app.py CHANGED
@@ -9,7 +9,7 @@ from model_utils import get_model
9
  MODEL = None
10
 
11
 
12
- def get_results(model_name: str, library: str, precision: list, training: list, access_token: str, zero_stage: int, num_nodes: int, num_gpus: int, offloading: list, zero_init: list, additional_buffer_factor: float):
13
  global MODEL
14
 
15
  MODEL = get_model(model_name, library, access_token)
 
9
  MODEL = None
10
 
11
 
12
+ def get_results(model_name: str, library: str, access_token: str):
13
  global MODEL
14
 
15
  MODEL = get_model(model_name, library, access_token)