Talking Head Anime from a Single ImageをCPUで動かしたい

Talking Head Anime from a Single Imageを非NVIDIAの製品で動かすために調べたところこのサイトのあるページにたどり着き書かれているとおりに

manual_poser.pyの

torch.device('cuda')
↓
torch.device('cpu')

util.pyの

def torch_load(file_name):
with open(file_name, 'rb') as f:
return torch.load(f)
を
def torch_load(file_name):
with open(file_name, 'rb') as f:
return torch.load(f, torch.device('cpu'))

とすれば動くのらしいのですが試したところ

起動して画像を選択すると

TypeError: torch_load() got an unexpected keyword argument 'map_location'

と出て右側に画像が表示されません。
どうすれば動くようになるでしょうか