Skip to content

Commit 18569f9

Browse files
committed
Remove device settings
1 parent 7eab46d commit 18569f9

File tree

1 file changed

+0
-9
lines changed

1 file changed

+0
-9
lines changed

beginner_source/t5_tutorial.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -20,14 +20,6 @@
2020
#
2121
#
2222

23-
######################################################################
24-
# Common imports
25-
# --------------
26-
import torch
27-
28-
DEVICE = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
29-
30-
3123
#######################################################################
3224
# Data Transformation
3325
# -------------------
@@ -86,7 +78,6 @@
8678
transform = t5_base.transform()
8779
model = t5_base.get_model()
8880
model.eval()
89-
model.to(DEVICE)
9081

9182

9283
#######################################################################

0 commit comments

Comments
 (0)