vahidthegreat
commited on
Commit
•
e294b35
1
Parent(s):
0df8b32
Update README.md
Browse files
README.md
CHANGED
@@ -63,7 +63,7 @@ class SiameseNetworkMPNet(nn.Module):
|
|
63 |
|
64 |
# Initialize with LoRA-applied model
|
65 |
self.model = AutoModel.from_pretrained(model_name)
|
66 |
-
self.model = PeftModel.from_pretrained(self.model, "
|
67 |
self.normalize = normalize
|
68 |
self.tokenizer = tokenizer
|
69 |
|
|
|
63 |
|
64 |
# Initialize with LoRA-applied model
|
65 |
self.model = AutoModel.from_pretrained(model_name)
|
66 |
+
self.model = PeftModel.from_pretrained(self.model, "vahidthegreat/StanceAware-SBERT")
|
67 |
self.normalize = normalize
|
68 |
self.tokenizer = tokenizer
|
69 |
|