Instructions to use Ashishkr/llama2-PII-Masking with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use Ashishkr/llama2-PII-Masking with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-hf") model = PeftModel.from_pretrained(base_model, "Ashishkr/llama2-PII-Masking") - Notebooks
- Google Colab
- Kaggle
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1!
#3
by vivekwagh2003 - opened
response = remove_pii_info(
model,
tokenizer,
prompt,256,
temperature=0.7)
getting the above error over here ....pls help