MOSS550V commited on
Commit
9268d17
·
1 Parent(s): f2d4d03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -148,7 +148,9 @@ def main():
148
 
149
  # P-tuning v2
150
  # model = model.half()
151
- model.transformer.prefix_encoder.float()
 
 
152
 
153
  model = model.eval()
154
  demo.queue().launch(share=False, inbrowser=True)
 
148
 
149
  # P-tuning v2
150
  # model = model.half()
151
+ # model.transformer.prefix_encoder.float()
152
+
153
+ model.float()
154
 
155
  model = model.eval()
156
  demo.queue().launch(share=False, inbrowser=True)