ford442 commited on
Commit
3739ae6
·
verified ·
1 Parent(s): 7a7c822

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -25,6 +25,7 @@ import datetime
25
  from gradio import themes
26
  from hidiffusion import apply_hidiffusion, remove_hidiffusion
27
  import gc
 
28
 
29
  torch.backends.cuda.matmul.allow_tf32 = False
30
  torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
 
25
  from gradio import themes
26
  from hidiffusion import apply_hidiffusion, remove_hidiffusion
27
  import gc
28
+ from diffusers.utils.torch_utils import randn_tensor
29
 
30
  torch.backends.cuda.matmul.allow_tf32 = False
31
  torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False