vision-rag-sourced-docker-image

Paused

manu commited on Mar 21

Commit

b78ca65

verified ·

1 Parent(s): ba29302

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -194,7 +194,7 @@ def index_gpu(images, ds):
     dataloader = DataLoader(
         images,
         batch_size=4,
-        num_workers=4,
         shuffle=False,
         collate_fn=lambda x: processor.process_images(x).to(model.device),
     )

     dataloader = DataLoader(
         images,
         batch_size=4,
+        # num_workers=4,
         shuffle=False,
         collate_fn=lambda x: processor.process_images(x).to(model.device),
     )