microsoft
/

LLM2CLIP-Llama-3.2-1B-Instruct-CC-Finetuned

Model card Files Files and versions

Gengzigang commited on Nov 30, 2024

Commit

889d606

·

1 Parent(s): 52e9e56

update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -29,7 +29,7 @@ class LlamaVec_1B_FeatureExtractor(nn.Module):
     def __init__(self):
         super().__init__()
-        model_path = 'LLM2CLIP-Llama-3-2-1B-Instruct-CC-Finetuned'
         config = AutoConfig.from_pretrained(model_path)
         model = AutoModel.from_pretrained(model_path, config=config, trust_remote_code=True, torch_dtype=torch.bfloat16)

     def __init__(self):
         super().__init__()
+        model_path = 'microsoft/LLM2CLIP-Llama-3.2-1B-Instruct-CC-Finetuned'
         config = AutoConfig.from_pretrained(model_path)
         model = AutoModel.from_pretrained(model_path, config=config, trust_remote_code=True, torch_dtype=torch.bfloat16)