Skip to content

Commit

Permalink
fix: Filepath in custom-deployment-template.yaml (#757)
Browse files Browse the repository at this point in the history
Fix the `inference_api.py` filepath used in
custom-deployment-template.yaml

#753

Signed-off-by: Ishaan Sehgal <ishaanforthewin@gmail.com>
  • Loading branch information
ishaansehgal99 authored Dec 4, 2024
1 parent dba607b commit fea2924
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/custom-model-integration/custom-deployment-template.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ inference:
- "1"
- "--gpu_ids"
- "all"
- "inference_api.py"
- "tfs/inference_api.py"
- "--pipeline"
- "text-generation"
- "--torch_dtype"
Expand All @@ -33,4 +33,4 @@ inference:
volumes:
- name: dshm
emptyDir:
medium: Memory
medium: Memory

0 comments on commit fea2924

Please sign in to comment.