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 |
|