We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7eab46d commit 18569f9Copy full SHA for 18569f9
beginner_source/t5_tutorial.py
@@ -20,14 +20,6 @@
20
#
21
22
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
31
#######################################################################
32
# Data Transformation
33
# -------------------
@@ -86,7 +78,6 @@
86
78
transform = t5_base.transform()
87
79
model = t5_base.get_model()
88
80
model.eval()
89
-model.to(DEVICE)
90
81
91
82
92
83
0 commit comments