Spaces:

misterbrainley
/

generate_dnd_images

Runtime error

misterbrainley commited on Jun 15, 2022

Commit

2496d9f

•

1 Parent(s): 7ea3fea

map saved model onto cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,9 @@ from utils import (
 N = 16
 cvae = ConditionalVariationalAutoEncoder()
-cvae.load_state_dict(torch.load('cvae.pth'))
 def generate(class_, species_, gender_):
     c_i = classes.index(class_)

 N = 16
 cvae = ConditionalVariationalAutoEncoder()
+# read state dict and map onto cpu
+cvae.load_state_dict(torch.load('cvae.pth', map_location=torch.device('cpu')))
 def generate(class_, species_, gender_):
     c_i = classes.index(class_)