From 8e86a1aa04636513f627ff603a998de1c462d2b9 Mon Sep 17 00:00:00 2001 From: Rucha Apte Date: Fri, 14 Feb 2025 14:46:08 -0800 Subject: [PATCH] Update tutorials/dapt-curation/code/main.py Co-authored-by: Sarah Yurick <53962159+sarahyurick@users.noreply.github.com> Signed-off-by: Rucha Apte --- tutorials/dapt-curation/code/main.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tutorials/dapt-curation/code/main.py b/tutorials/dapt-curation/code/main.py index f1941fda..25a1e729 100755 --- a/tutorials/dapt-curation/code/main.py +++ b/tutorials/dapt-curation/code/main.py @@ -277,7 +277,6 @@ def main(): args = ArgumentHelper(parser).add_distributed_args().parse_args() # Limit the total number of workers to ensure we don't run out of memory. args.n_workers = min(args.n_workers, 8) - args.device = "gpu" print("Args: ", args) # Download all the sources and get the list of text and code files.