hujiecpp commited on
Commit
da2ad26
·
1 Parent(s): 607d75e

init project

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -304,8 +304,8 @@ def get_cog_feats(images):
304
 
305
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
306
  pe3r.sam2.to(device)
307
- pe3r.siglip_processor.to(device)
308
- pe3r.siglip.to(device)
309
 
310
  cog_seg_maps = []
311
  rev_cog_seg_maps = []
@@ -518,7 +518,7 @@ def get_3D_object_from_scene(outdir, text, threshold, scene, min_conf_thr, as_po
518
  mask_sky, clean_depth, transparent_cams, cam_size):
519
 
520
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
521
- pe3r.siglip_tokenizer.to(device)
522
 
523
 
524
  texts = [text]
 
304
 
305
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
306
  pe3r.sam2.to(device)
307
+ # pe3r.siglip_processor.to(device)
308
+ # pe3r.siglip.to(device)
309
 
310
  cog_seg_maps = []
311
  rev_cog_seg_maps = []
 
518
  mask_sky, clean_depth, transparent_cams, cam_size):
519
 
520
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
521
+ # pe3r.siglip_tokenizer.to(device)
522
 
523
 
524
  texts = [text]