Commit ·
73048e5
1
Parent(s): f9adc3c
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() 使用GPU
|
| 48 |
|
| 49 |
-
model = model.float() # 使用CPU
|
| 50 |
model.transformer.prefix_encoder.float()
|
| 51 |
model = model.eval()
|
| 52 |
|
|
|
|
| 46 |
|
| 47 |
#model = model.half().cuda() 使用GPU
|
| 48 |
|
| 49 |
+
model = model.half().float() # 使用CPU
|
| 50 |
model.transformer.prefix_encoder.float()
|
| 51 |
model = model.eval()
|
| 52 |
|