Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -40,12 +40,12 @@ from gradio import themes
|
|
40 |
from image_gen_aux import UpscaleWithModel
|
41 |
#from diffusers.models.attention_processor import AttnProcessor2_0
|
42 |
|
43 |
-
torch.backends.cuda.matmul.allow_tf32 =
|
44 |
torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
|
45 |
torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
|
46 |
-
torch.backends.cudnn.allow_tf32 =
|
47 |
torch.backends.cudnn.deterministic = False
|
48 |
-
torch.backends.cudnn.benchmark =
|
49 |
torch.backends.cuda.preferred_blas_library="cublas"
|
50 |
torch.backends.cuda.preferred_linalg_library="cusolver"
|
51 |
torch.set_float32_matmul_precision("highest")
|
|
|
40 |
from image_gen_aux import UpscaleWithModel
|
41 |
#from diffusers.models.attention_processor import AttnProcessor2_0
|
42 |
|
43 |
+
torch.backends.cuda.matmul.allow_tf32 = False
|
44 |
torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
|
45 |
torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
|
46 |
+
torch.backends.cudnn.allow_tf32 = False
|
47 |
torch.backends.cudnn.deterministic = False
|
48 |
+
torch.backends.cudnn.benchmark = False
|
49 |
torch.backends.cuda.preferred_blas_library="cublas"
|
50 |
torch.backends.cuda.preferred_linalg_library="cusolver"
|
51 |
torch.set_float32_matmul_precision("highest")
|