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

Question

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'

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

Comments

[jbpb0]

https://detail.chiebukuro.yahoo.co.jp/qa/question_detail/q11268023781?__ysp=dG9yY2hfbG9hZA%3D%3D
を投稿したのは質問者さんでしょうか？

もしそうなら、
https://teratail.com/help#posted-otherservice
を見てください

[nekusu]

複数使うのってだめなんですね。
向こうは回答出そうになさそうなので消しました

[jbpb0]

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

よりも上に「Traceback」と書かれてたら、それ以降を全部質問に記載した方が、何が起きてるか他人が把握しやすくなります
(ここに書くのではなく、質問を編集して追記する)

ユーザー名等の個人情報は伏せ字にしていいですが、それ以外はできるだけ省略しないで記載した方がいいです

Answer 1

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

エラーからすると、提示されたコードにないところで、torch_loadが不正な引数で呼ばれているようですね。