Update README.md
Browse files
README.md
CHANGED
@@ -53,8 +53,8 @@ model = transformers.AutoModelForCausalLM.from_pretrained(
|
|
53 |
use_auth_token=hf_auth
|
54 |
)
|
55 |
|
56 |
-
config = PeftConfig.from_pretrained("Ashishkr/
|
57 |
-
model = PeftModel.from_pretrained(model, "Ashishkr/
|
58 |
|
59 |
model.eval()
|
60 |
print(f"Model loaded on {device}")
|
|
|
53 |
use_auth_token=hf_auth
|
54 |
)
|
55 |
|
56 |
+
config = PeftConfig.from_pretrained("Ashishkr/PII-Masking")
|
57 |
+
model = PeftModel.from_pretrained(model, "Ashishkr/PII-Masking").to(device)
|
58 |
|
59 |
model.eval()
|
60 |
print(f"Model loaded on {device}")
|