Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
500319a
1
Parent(s):
59a2caa
adding natten import test
Browse files
app.py
CHANGED
|
@@ -25,6 +25,9 @@ from tqdm import trange, tqdm
|
|
| 25 |
from torchvision import transforms
|
| 26 |
import k_diffusion as K
|
| 27 |
|
|
|
|
|
|
|
|
|
|
| 28 |
zero = torch.Tensor([0]).cuda()
|
| 29 |
print("Zero Device = ",zero.device," <-- this probably says cpu") # <-- 'cpu' 🤔
|
| 30 |
|
|
|
|
| 25 |
from torchvision import transforms
|
| 26 |
import k_diffusion as K
|
| 27 |
|
| 28 |
+
# test natten import:
|
| 29 |
+
import natten
|
| 30 |
+
|
| 31 |
zero = torch.Tensor([0]).cuda()
|
| 32 |
print("Zero Device = ",zero.device," <-- this probably says cpu") # <-- 'cpu' 🤔
|
| 33 |
|