Optimize compatibility other devices (like cpu)

pkhungurn / talking-head-anime-demo

Demo for the "Talking Head Anime from a Single Image."

MIT License

1.99k stars 286 forks source link

Closed sheiun closed 4 years ago

sheiun commented 4 years ago

Pass other device in __main__ block

device = torch.device("cpu")
poser = MorphRotateCombinePoser256Param6(
    ...
    device=device)

sheiun commented 4 years ago

Tested on Macbook Air 13" 2018