File tree Expand file tree Collapse file tree
evaluation_function/models Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -104,4 +104,5 @@ jobs:
104104 aws-key-id : ${{ secrets.LAMBDA_CONTAINER_PIPELINE_AWS_ID }}
105105 aws-secret-key : ${{ secrets.LAMBDA_CONTAINER_PIPELINE_AWS_SECRET}}
106106 function-admin-api-key : ${{ secrets.FUNCTION_ADMIN_API_KEY}}
107+ gcp_credentials : ${{ secrets.GCP_DEPLOY }}
107108
Original file line number Diff line number Diff line change @@ -72,10 +72,8 @@ def train_model(device):
7272 return model
7373
7474def run (response , answer , params : Params ) -> Result :
75- # print("GPU") if torch.backends.mps.is_available() else print("CPU")
76- # device = torch.device("mps" if torch.backends.mps.is_available() else "cpu")
77-
78- device = "cpu"
75+ print ("GPU" ) if torch .backends .mps .is_available () else print ("CPU" )
76+ device = torch .device ("mps" if torch .backends .mps .is_available () else "cpu" )
7977
8078 refresh = params .get ("refresh" , False )
8179 if refresh :
You can’t perform that action at this time.
0 commit comments