Spaces:
Sleeping
Sleeping
mkaraki
commited on
Commit
•
0e03fc1
1
Parent(s):
2d200d3
Change code for huggingface
Browse files
app.py
CHANGED
|
@@ -94,7 +94,7 @@ if (ngpu > 1) and (device.type == 'cuda'):
|
|
| 94 |
netG.apply(weights_init)
|
| 95 |
|
| 96 |
|
| 97 |
-
checkpoint = torch.load("checkpoints/epoch1100.ckpt")
|
| 98 |
|
| 99 |
|
| 100 |
if omit_module:
|
|
|
|
| 94 |
netG.apply(weights_init)
|
| 95 |
|
| 96 |
|
| 97 |
+
checkpoint = torch.load("checkpoints/epoch1100.ckpt", map_location=torch.device('cpu'))
|
| 98 |
|
| 99 |
|
| 100 |
if omit_module:
|