Open cip22 opened 3 months ago
Yeah, similar issue happened in Mac book M1 pro, keep crashing Jupiter notebook when running the pipeline. it works after specified GPU acceleration: nlp = pipeline("sentiment-analysis", framework='pt',device=0). device=0 means using GPU, -1 means use CPU, default value is -1.
Interesting, so there seems to be more than one way to specify the device.
When I try to run the sentiment analysis, as soon as I execute
nlp("I feel good")
, I getThe problem turned out to be this code:
nlp = pipeline("sentiment-analysis")
After this step, I get - among other info - this line back:So I decided to provide the 'device' argument. If MPS is present on the machine, I can create the pipeline like this:
nlp = pipeline("sentiment-analysis", device="mps")
The documentation is here: https://huggingface.co/docs/transformers/v4.44.0/en/main_classes/pipelines#transformers.pipelineNow the sentiment analysis runs.