Spaces:

nyanko7
/

toaru-xl-model

Running on Zero

nyanko7 commited on about 3 hours ago

Commit

a2e7469

•

1 Parent(s): b4181ad

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1727,7 +1727,7 @@ pipe2 = StableDiffusionXLPipeline.from_single_file(
     torch_dtype=torch.float16
 )
 pipe2.fuse_qkv_projections()
-pipe2.scheduler = FlowMatchEulerDiscreteScheduler(num_train_timesteps=1000, shift=2.0)
 pipe2.scheduler.init_noise_sigma = 1.0
 pipe2.scheduler.scale_model_input = lambda x, y: x
 pipe2.unet.to(memory_format=torch.channels_last)

     torch_dtype=torch.float16
 )
 pipe2.fuse_qkv_projections()
+pipe2.scheduler = FlowMatchEulerDiscreteScheduler(num_train_timesteps=1000, shift=1.5)
 pipe2.scheduler.init_noise_sigma = 1.0
 pipe2.scheduler.scale_model_input = lambda x, y: x
 pipe2.unet.to(memory_format=torch.channels_last)