lucas-w commited on
Commit
2cc3545
β€’
1 Parent(s): ffe98a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -64,6 +64,10 @@ def givetext(input_text,lmodel,ltokenizer):
64
  print("BEFORE ATTEMPTING TO MOVE LMODEL TO CPU")
65
 
66
  lmodel = lmodel.to("cpu")
 
 
 
 
67
 
68
  print("BEFORE GENERATING LMODEL")
69
 
 
64
  print("BEFORE ATTEMPTING TO MOVE LMODEL TO CPU")
65
 
66
  lmodel = lmodel.to("cpu")
67
+
68
+ print("BEFORE ATTEMPTING .cpu()")
69
+
70
+ lmodel.cpu()
71
 
72
  print("BEFORE GENERATING LMODEL")
73