mustapha commited on
Commit
b74d347
1 Parent(s): 9ae79ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ model = PeftModel.from_pretrained(
30
  )
31
 
32
  device = "cpu"
33
- print("model device :", model.device)
34
  # model.to(device)
35
  model.eval()
36
 
 
30
  )
31
 
32
  device = "cpu"
33
+ print("model device :", model.device, flush=True)
34
  # model.to(device)
35
  model.eval()
36