Spaces:
Sleeping
Sleeping
mkaraki
commited on
Commit
·
d406a66
1
Parent(s):
5e892ae
Change code for huggingface
Browse files
app.py
CHANGED
|
@@ -35,7 +35,7 @@ lr = 0.0002
|
|
| 35 |
beta1 = 0.5
|
| 36 |
|
| 37 |
# Number of GPUs available. Use 0 for CPU mode.
|
| 38 |
-
ngpu =
|
| 39 |
|
| 40 |
|
| 41 |
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
|
|
|
|
| 35 |
beta1 = 0.5
|
| 36 |
|
| 37 |
# Number of GPUs available. Use 0 for CPU mode.
|
| 38 |
+
ngpu = 0
|
| 39 |
|
| 40 |
|
| 41 |
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
|