Remove .cuda() from cm3leon's model.py (#1897)

Summary: Also remove the `.eval()` as the `.eval()` and the moving to device is done immediately afterward in the model's `__init__.py` Pull Request resolved: #1897 Reviewed By: msaroufim Differential Revision: D49285752 Pulled By: xuzhao9 fbshipit-source-id: 441b01c63ee9b48b792dea121e337fce1b2e330b
pytorch · Sep 15, 2023 · 6532c61 · 6532c61
1 parent e1b51f5
commit 6532c61
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/torchbenchmark/models/cm3leon_generate/model.py b/torchbenchmark/models/cm3leon_generate/model.py
@@ -1196,6 +1196,4 @@ def create_model(embed_dim=1536):
             decoder_learned_pos=True,
             dropout=0.1,
         )
-        .eval()
-        .cuda()
     )