banao-tech commited on
Commit
3d8571e
·
verified ·
1 Parent(s): 52027db

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -44,13 +44,13 @@ processor = AutoProcessor.from_pretrained(
44
 
45
  try:
46
  model = AutoModelForCausalLM.from_pretrained(
47
- "microsoft/OmniParser",
48
  torch_dtype=torch.float16,
49
  trust_remote_code=True,
50
  ).to("cuda")
51
  except:
52
  model = AutoModelForCausalLM.from_pretrained(
53
- "microsoft/OmniParser",
54
  torch_dtype=torch.float16,
55
  trust_remote_code=True,
56
  )
 
44
 
45
  try:
46
  model = AutoModelForCausalLM.from_pretrained(
47
+ "banao-tech/OmniParse",
48
  torch_dtype=torch.float16,
49
  trust_remote_code=True,
50
  ).to("cuda")
51
  except:
52
  model = AutoModelForCausalLM.from_pretrained(
53
+ "banao-tech/OmniParse",
54
  torch_dtype=torch.float16,
55
  trust_remote_code=True,
56
  )