SituatedEmbedding
/

SitEmb-v1.5-Qwen3-note

Sentence Similarity

Model card Files Files and versions

SituatedEmbedding commited on Aug 3

Commit

d41868f

·

verified ·

1 Parent(s): 106eed5

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ tokenizer = AutoTokenizer.from_pretrained(
 model = AutoModel.from_pretrained(
     "SituatedEmbedding/SitEmb-v1.5-Qwen3-note",
-    torch_dtype=torch.bfloat16 if args.bf16 else torch.float32,
     device_map={"": 0},
 )

 model = AutoModel.from_pretrained(
     "SituatedEmbedding/SitEmb-v1.5-Qwen3-note",
+    torch_dtype=torch.bfloat16,
     device_map={"": 0},
 )