Spaces:
Runtime error
Runtime error
Gustavo Belfort
commited on
Commit
•
3ac23fa
1
Parent(s):
7b31381
change interface
Browse files- interface.py +1 -1
interface.py
CHANGED
|
@@ -19,7 +19,7 @@ with open(cached_download(hf_hub_url('ykilcher/apes', 'gamma500/network-snapshot
|
|
| 19 |
G = pickle.load(f)["G_ema"] # torch.nn.Module
|
| 20 |
|
| 21 |
device = torch.device("cpu")
|
| 22 |
-
if
|
| 23 |
device = torch.device("cuda")
|
| 24 |
G = G.to(device)
|
| 25 |
else:
|
|
|
|
| 19 |
G = pickle.load(f)["G_ema"] # torch.nn.Module
|
| 20 |
|
| 21 |
device = torch.device("cpu")
|
| 22 |
+
if false:
|
| 23 |
device = torch.device("cuda")
|
| 24 |
G = G.to(device)
|
| 25 |
else:
|