Skip to content

Commit 3277a39

Browse files
committed
Added GCP secrets
1 parent 93e692e commit 3277a39

2 files changed

Lines changed: 3 additions & 4 deletions

File tree

.github/workflows/deploy.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff 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

evaluation_function/models/basic_nn.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -72,10 +72,8 @@ def train_model(device):
7272
return model
7373

7474
def 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:

0 commit comments

Comments
 (0)