dongdongzi commited on
Commit
6bd5b30
·
verified ·
1 Parent(s): 8177c52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,8 +19,8 @@ cpu_checkpoint_path = "efficientsam_s_cpu.jit"
19
  ti_gpu_checkpoint_path = "efficientsam_ti_gpu.jit"
20
 
21
  if torch.cuda.is_available():
22
- model = torch.jit.load(ti_gpu_checkpoint_path)
23
- # model = torch.jit.load(gpu_checkpoint_path)
24
  print(f"using model {ti_gpu_checkpoint_path}")
25
  else:
26
  model = torch.jit.load(cpu_checkpoint_path)
 
19
  ti_gpu_checkpoint_path = "efficientsam_ti_gpu.jit"
20
 
21
  if torch.cuda.is_available():
22
+ #model = torch.jit.load(ti_gpu_checkpoint_path)
23
+ model = torch.jit.load(gpu_checkpoint_path)
24
  print(f"using model {ti_gpu_checkpoint_path}")
25
  else:
26
  model = torch.jit.load(cpu_checkpoint_path)