banao-tech commited on
Commit
0141f51
·
verified ·
1 Parent(s): 1ec8443

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -3
main.py CHANGED
@@ -41,9 +41,7 @@ class ModelManager:
41
  )
42
 
43
  self.model = Blip2ForConditionalGeneration.from_pretrained("banao-tech/OmniParser",torch_dtype=torch.float16,
44
- trust_remote_code=True)
45
-
46
- ).to(self.device)
47
 
48
  return True
49
  except Exception as e:
 
41
  )
42
 
43
  self.model = Blip2ForConditionalGeneration.from_pretrained("banao-tech/OmniParser",torch_dtype=torch.float16,
44
+ trust_remote_code=True).to(self.device)
 
 
45
 
46
  return True
47
  except Exception as e: