toncho11 commited on
Commit
5653788
1 Parent(s): 8ee988b

Fixes an error when running on CPU mode.

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -76,7 +76,8 @@ def generate_prompt(instruction, input=None):
76
 
77
  ### Response:"""
78
 
79
- model.half()
 
80
  model.eval()
81
  if torch.__version__ >= "2":
82
  model = torch.compile(model)
 
76
 
77
  ### Response:"""
78
 
79
+ if device != "cpu":
80
+ model.half()
81
  model.eval()
82
  if torch.__version__ >= "2":
83
  model = torch.compile(model)