diff --git a/github_issue_summarization/pipelines/example_pipelines/gh_summ.py b/github_issue_summarization/pipelines/example_pipelines/gh_summ.py
index 0398ae2c4..bf0bde1a1 100644
--- a/github_issue_summarization/pipelines/example_pipelines/gh_summ.py
+++ b/github_issue_summarization/pipelines/example_pipelines/gh_summ.py
@@ -95,7 +95,7 @@ def gh_summ(  #pylint: disable=unused-argument
   train.after(copydata)
   log_model.after(train)
   serve.after(train)
-  train.set_gpu_limit(4)
+  train.set_gpu_limit(1)
   train.set_memory_limit('48G')
 
   with dsl.Condition(train.output == 'true'):
diff --git a/github_issue_summarization/pipelines/example_pipelines/gh_summ.py.tar.gz b/github_issue_summarization/pipelines/example_pipelines/gh_summ.py.tar.gz
index f9a7a864a..b285e71f4 100644
Binary files a/github_issue_summarization/pipelines/example_pipelines/gh_summ.py.tar.gz and b/github_issue_summarization/pipelines/example_pipelines/gh_summ.py.tar.gz differ