Spaces:

Shriharshan
/

Autism-RAG

Running

Shriharshan commited on Mar 26

Commit

5843ebf

verified ·

1 Parent(s): 43d3036

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,6 @@ from langchain.docstore.document import Document
 model = HuggingFacePipeline.from_model_id(
     model_id="HuggingFaceTB/SmolLM2-360M-Instruct",
     task="text-generation",
-    device_map="auto",
     pipeline_kwargs=dict(
         max_new_tokens=512,
         do_sample=False,

 model = HuggingFacePipeline.from_model_id(
     model_id="HuggingFaceTB/SmolLM2-360M-Instruct",
     task="text-generation",
     pipeline_kwargs=dict(
         max_new_tokens=512,
         do_sample=False,