Hugging Face's logo

Spaces:
yuntian-group
/
neural-os
Runtime error

App Files Community
da03 commited on
Commit
042c554
·
1 Parent(s): 3a94a55

.

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -14,8 +14,8 @@ import time
14
  from typing import Any, Dict
15
  from ldm.models.diffusion.ddpm import LatentDiffusion, DDIMSampler
16
 
17
- torch.backends.cuda.matmul.allow_tf32 = True
18
- torch.backends.cudnn.allow_tf32 = True
19
  SCREEN_WIDTH = 512
20
  SCREEN_HEIGHT = 384
21
  NUM_SAMPLING_STEPS = 8
 
14
  from typing import Any, Dict
15
  from ldm.models.diffusion.ddpm import LatentDiffusion, DDIMSampler
16
 
17
+ #torch.backends.cuda.matmul.allow_tf32 = True
18
+ #torch.backends.cudnn.allow_tf32 = True
19
  SCREEN_WIDTH = 512
20
  SCREEN_HEIGHT = 384
21
  NUM_SAMPLING_STEPS = 8