Skip to content

Commit

Permalink
chore: Update gpu flag in predict function to use request value
Browse files Browse the repository at this point in the history
  • Loading branch information
amirnd51 committed Jul 24, 2024
1 parent 6f310c9 commit 95fc212
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python_api/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -319,7 +319,7 @@ async def predict(request: PredictRequest):
architecture = request.architecture
batch_size = request.batchSize
desired_result_modality = request.desiredResultModality
gpu = True
gpu = request.gpu
inputs = request.inputs
# input_url = request.input_url
model_id = request.model
Expand Down

0 comments on commit 95fc212

Please sign in to comment.