Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -1727,7 +1727,7 @@ pipe2 = StableDiffusionXLPipeline.from_single_file(
|
|
1727 |
torch_dtype=torch.float16
|
1728 |
)
|
1729 |
pipe2.fuse_qkv_projections()
|
1730 |
-
pipe2.scheduler = FlowMatchEulerDiscreteScheduler(num_train_timesteps=1000, shift=
|
1731 |
pipe2.scheduler.init_noise_sigma = 1.0
|
1732 |
pipe2.scheduler.scale_model_input = lambda x, y: x
|
1733 |
pipe2.unet.to(memory_format=torch.channels_last)
|
|
|
1727 |
torch_dtype=torch.float16
|
1728 |
)
|
1729 |
pipe2.fuse_qkv_projections()
|
1730 |
+
pipe2.scheduler = FlowMatchEulerDiscreteScheduler(num_train_timesteps=1000, shift=1.5)
|
1731 |
pipe2.scheduler.init_noise_sigma = 1.0
|
1732 |
pipe2.scheduler.scale_model_input = lambda x, y: x
|
1733 |
pipe2.unet.to(memory_format=torch.channels_last)
|