Skip to content

Device to use (e.g. cpu, cuda:0, cuda:1, etc.)#1844

Merged
patil-suraj merged 2 commits into
huggingface:mainfrom
camenduru:map_location
Dec 27, 2022
Merged

Device to use (e.g. cpu, cuda:0, cuda:1, etc.)#1844
patil-suraj merged 2 commits into
huggingface:mainfrom
camenduru:map_location

"cuda" if torch.cuda.is_available() else "cpu"

ea6c517
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs