SimSwap_Device_Agnositc_Patch.bat

<# :
@echo off
powershell -noprofile -NoLogo "iex (${%~f0} | out-string)"
: #>
$search_replace = @{
    "if len(self.opt.gpu_ids)" = "if torch.cuda.is_available() and len(self.opt.gpu_ids)";
    "torch.device(`'cuda:0`')" = "torch.device(`'cuda:0`' if torch.cuda.is_available() else `'cpu`')";
    "torch.load(netArc_checkpoint)" = "torch.load(netArc_checkpoint) if torch.cuda.is_available() else torch.load(netArc_checkpoint, map_location=torch.device(`'cpu`'))";
    ".cuda()" = ".to(torch.device(`'cuda:0`' if torch.cuda.is_available() else `'cpu`'))";
    ".to('cuda')" = ".to(torch.device(`'cuda:0`' if torch.cuda.is_available() else `'cpu`'))";
}

ForEach ($File in (Get-ChildItem -Path '.\*.py' -Recurse -File)) {
    $content = (Get-Content $File)
    if ($content.length -gt 0) {
        ForEach ($search in $search_replace.Keys) {
            $content = $content.replace("$search", "$($search_replace[$search])")
        }
        Set-Content $File $content
    }
}