Update tutorials/dapt-curation/code/main.py

Co-authored-by: Sarah Yurick <[email protected]> Signed-off-by: Rucha Apte <[email protected]>
NVIDIA · Feb 14, 2025 · 8e86a1a · 8e86a1a
1 parent dab3789
commit 8e86a1a
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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.