kevinwang676
commited on
Commit
•
83fc5d0
1
Parent(s):
75b94c9
Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +44,10 @@ for k, v in prefix_state_dict.items():
|
|
44 |
new_prefix_state_dict[k[len("transformer.prefix_encoder."):]] = v
|
45 |
model.transformer.prefix_encoder.load_state_dict(new_prefix_state_dict)
|
46 |
|
47 |
-
model = model.half().cuda()
|
|
|
|
|
|
|
48 |
model.transformer.prefix_encoder.float()
|
49 |
model = model.eval()
|
50 |
|
|
|
44 |
new_prefix_state_dict[k[len("transformer.prefix_encoder."):]] = v
|
45 |
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()
|
53 |
|