8000 Cant use mulitple GPUs · Issue #1 · enginetix2/ragllm2 · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
Cant use mulitple GPUs #1
New issue
New issue
Open
Open
Cant use mulitple GPUs#1
@enginetix2

Description

@enginetix2
enginetix2
opened on Mar 25, 2024

embeddings = HuggingFaceEmbeddings(model_name="BAAI/bge-large-en-v1.5", show_progress=True, model_kwargs={'device': "cuda:0"})

This is the only working embbedings gpu call

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions
      0