Spaces:

manabb
/

WithclickablePDF

Sleeping

manabb commited on Dec 28, 2025

Commit

fec2475

verified ·

1 Parent(s): e181aa6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -163,7 +163,7 @@ def create_llm_pipeline():
     return HuggingFacePipeline.from_model_id(
         model_id="microsoft/DialoGPT-medium",
         task="text-generation",
-        device_map="auto" if torch.cuda.is_available() else device_map = 'cpu',
         pipeline_kwargs={
             "max_new_tokens": 200,
             "do_sample": True,

     return HuggingFacePipeline.from_model_id(
         model_id="microsoft/DialoGPT-medium",
         task="text-generation",
+        device_map="auto",
         pipeline_kwargs={
             "max_new_tokens": 200,
             "do_sample": True,