baohuynhbk14 commited on
Commit
3d228e4
·
verified ·
1 Parent(s): 7dcfec3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -182,11 +182,6 @@ import matplotlib.pyplot as plt
182
  # ==============================
183
  model_name = "5CD-AI/Vintern-Embedding-1B"
184
 
185
- processor = AutoProcessor.from_pretrained(
186
- model_name,
187
- trust_remote_code=True
188
- )
189
-
190
  model = AutoModel.from_pretrained(
191
  model_name,
192
  torch_dtype=torch.bfloat16,
@@ -194,6 +189,11 @@ model = AutoModel.from_pretrained(
194
  trust_remote_code=True,
195
  ).eval().cuda()
196
 
 
 
 
 
 
197
  # ==============================
198
  # 2. Prepare Input Data
199
  # ==============================
 
182
  # ==============================
183
  model_name = "5CD-AI/Vintern-Embedding-1B"
184
 
 
 
 
 
 
185
  model = AutoModel.from_pretrained(
186
  model_name,
187
  torch_dtype=torch.bfloat16,
 
189
  trust_remote_code=True,
190
  ).eval().cuda()
191
 
192
+ processor = AutoProcessor.from_pretrained(
193
+ model_name,
194
+ trust_remote_code=True
195
+ )
196
+
197
  # ==============================
198
  # 2. Prepare Input Data
199
  # ==============================