Spaces:
Running
on
Zero
Running
on
Zero
Kohaku-Blueleaf
commited on
Commit
·
dba151a
1
Parent(s):
9746259
typo
Browse files
diff.py
CHANGED
@@ -5,7 +5,7 @@ from diffusers import StableDiffusionXLKDiffusionPipeline
|
|
5 |
from k_diffusion.sampling import get_sigmas_polyexponential
|
6 |
from k_diffusion.sampling import sample_dpmpp_2m_sde
|
7 |
|
8 |
-
torch.set_float32_matmul_precision("
|
9 |
|
10 |
|
11 |
def set_timesteps_polyexponential(self, orig_sigmas, num_inference_steps, device=None):
|
|
|
5 |
from k_diffusion.sampling import get_sigmas_polyexponential
|
6 |
from k_diffusion.sampling import sample_dpmpp_2m_sde
|
7 |
|
8 |
+
torch.set_float32_matmul_precision("medium")
|
9 |
|
10 |
|
11 |
def set_timesteps_polyexponential(self, orig_sigmas, num_inference_steps, device=None):
|