kevinwang676
commited on
Commit
•
bd158ce
1
Parent(s):
bc670cc
Update app.py
Browse files
app.py
CHANGED
@@ -46,7 +46,7 @@ model.transformer.prefix_encoder.load_state_dict(new_prefix_state_dict)
|
|
46 |
|
47 |
#model = model.half().cuda()
|
48 |
|
49 |
-
model = model.float()
|
50 |
|
51 |
model.transformer.prefix_encoder.float()
|
52 |
model = model.eval()
|
|
|
46 |
|
47 |
#model = model.half().cuda()
|
48 |
|
49 |
+
model = model.half().float()
|
50 |
|
51 |
model.transformer.prefix_encoder.float()
|
52 |
model = model.eval()
|