Spaces:

svjack
/

Question-Generator-on-Chinese-Doc

Sleeping

svjack commited on Oct 17, 2023

Commit

84bad2a

1 Parent(s): 9ee67e4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,7 @@ if torch.cuda.is_available():
     zh_pretrained_model.to(device)
 else:
     import torch
-    zh_pretrained_model.to(torch.bfloat16)
 #en_pretrained_model.eval()

     zh_pretrained_model.to(device)
 else:
     import torch
+    zh_pretrained_model.encoder.to(torch.bfloat16)
 #en_pretrained_model.eval()