Spaces:

JiakaiDu
/

RAG_Test

Paused

JiakaiDu commited on Sep 27

Commit

de9204e

•

1 Parent(s): 5df3113

Upload folder using huggingface_hub

Files changed (1) hide show

Test_RAG.py CHANGED Viewed

@@ -288,7 +288,7 @@ llm = HuggingFacePipeline.from_model_id(
     task="text-generation",
     backend="openvino",
     model_kwargs={
-        "device": llm_device.value,
         "ov_config": ov_config,
         "trust_remote_code": True,
     },

     task="text-generation",
     backend="openvino",
     model_kwargs={
+        "device": llm_device,
         "ov_config": ov_config,
         "trust_remote_code": True,
     },