Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|