Ability to specify a GPU index - #193

Open

vladkalinichencko wants to merge 5 commits into

facebookresearch:mainfrom

vladkalinichencko:main

Commits on Jan 13, 2024

Update device.py
vladkalinichencko
authored
f94ef07
Update predict.py
vladkalinichencko
authored
80c5980
removed comments
vladkalinichencko
committed
975dfa1
Updated with the new argument
vladkalinichencko
committed
6047fc5
removed strict=False arguments
vladkalinichencko
committed
9dec28f