haystack

Форк
0
/
sentence-transformers-embedders-new-devices-mgmt-07cb59c6b4a13280.yaml 
20 строк · 669.0 Байт
1
---
2
upgrade:
3
  - |
4
    Adopt the new framework-agnostic device management in Sentence Transformers Embedders.
5

6
    Before this change:
7
    ```python
8
    from haystack.components.embedders import SentenceTransformersTextEmbedder
9
    embedder = SentenceTransformersTextEmbedder(device="cuda:0")
10
    ```
11

12
    After this change:
13
    ```python
14
    from haystack.utils.device import ComponentDevice, Device
15
    from haystack.components.embedders import SentenceTransformersTextEmbedder
16
    device = ComponentDevice.from_single(Device.gpu(id=0))
17
    # or
18
    # device = ComponentDevice.from_str("cuda:0")
19
    embedder = SentenceTransformersTextEmbedder(device=device)
20
    ```
21

Использование cookies

Мы используем файлы cookie в соответствии с Политикой конфиденциальности и Политикой использования cookies.

Нажимая кнопку «Принимаю», Вы даете АО «СберТех» согласие на обработку Ваших персональных данных в целях совершенствования нашего веб-сайта и Сервиса GitVerse, а также повышения удобства их использования.

Запретить использование cookies Вы можете самостоятельно в настройках Вашего браузера.