diff --git a/examples/text-generator/cortex.yaml b/examples/text-generator/cortex.yaml index 1eeafc65f7..6fa55a7297 100644 --- a/examples/text-generator/cortex.yaml +++ b/examples/text-generator/cortex.yaml @@ -6,5 +6,5 @@ model: s3://cortex-examples/text-generator/gpt-2/124M request_handler: handler.py compute: + cpu: 1 gpu: 1 - cpu: 2