Spaces:

Muhammadidrees
/

RiayatechChatDoctor

Sleeping

Muhammadidrees commited on Sep 30

Commit

16e90de

verified ·

1 Parent(s): 4570e73

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,6 @@ def load_model():
         torch_dtype=dtype,
         device_map="auto"  # Let HF handle device placement
     )
-    model.to(device)
     return processor, model, device
 # Load model once at startup

         torch_dtype=dtype,
         device_map="auto"  # Let HF handle device placement
     )
     return processor, model, device
 # Load model once at startup