zxx_test
/
app_demo_test_1
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

Update app.py

This commit is contained in:
Baize 2023-04-04 03:55:24 +00:00 committed by huggingface-web
parent 85f3d2d553
commit ed54557c50
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app.py
View File

@ -41,7 +41,7 @@ def predict(text,
else: else:
prompt,inputs=inputs prompt,inputs=inputs
begin_length = len(prompt) begin_length = len(prompt)
torch.cuda.empty_cache()
input_ids = inputs["input_ids"].to(device) input_ids = inputs["input_ids"].to(device)
with torch.no_grad(): with torch.no_grad():