Flourish commited on
Commit
7d5c2c8
·
verified ·
1 Parent(s): 72c16f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -63,15 +63,11 @@ print("init ovis")
63
  # token=hf_token,
64
  # )
65
  ovis_tokenizer = build_ovis_tokenizer(
66
- "AIDC-AI/Ovis-Image-7B",
67
- subfolder="Ovis2.5-2B",
68
- token=hf_token,
69
  )
70
  ovis_encoder = OvisEmbedder(
71
- model_path="AIDC-AI/Ovis-Image-7B",
72
  random_init=False,
73
- subfolder="Ovis2.5-2B",
74
- token=hf_token,
75
  low_cpu_mem_usage=True,
76
  torch_dtype=torch.bfloat16,
77
  ).to(device=device, dtype=_dtype)
 
63
  # token=hf_token,
64
  # )
65
  ovis_tokenizer = build_ovis_tokenizer(
66
+ "AIDC-AI/Ovis2.5-2B",
 
 
67
  )
68
  ovis_encoder = OvisEmbedder(
69
+ model_path="AIDC-AI/Ovis2.5-2B",
70
  random_init=False,
 
 
71
  low_cpu_mem_usage=True,
72
  torch_dtype=torch.bfloat16,
73
  ).to(device=device, dtype=_dtype)