haystack
20 строк · 669.0 Байт
1---
2upgrade:3- |4Adopt the new framework-agnostic device management in Sentence Transformers Embedders.
5
6Before this change:7```python
8from haystack.components.embedders import SentenceTransformersTextEmbedder
9embedder = SentenceTransformersTextEmbedder(device="cuda:0")10```
11
12After this change:13```python
14from haystack.utils.device import ComponentDevice, Device15from haystack.components.embedders import SentenceTransformersTextEmbedder
16device = ComponentDevice.from_single(Device.gpu(id=0))
17# or18# device = ComponentDevice.from_str("cuda:0")19embedder = SentenceTransformersTextEmbedder(device=device)
20```
21