AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Change one-step dummy pipeline for testing (#1690)

Browse Source 
Change the one-step dummy pipeline for testing
This commit is contained in:
Patrick von Platen
2022-12-13 16:55:49 +01:00
committed by GitHub
parent 07f95503e5
commit e62dd5cfa8
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/community/one_step_unet.py
View File

@@ -18,5 +18,7 @@ class UnetSchedulerOneForwardPipeline(DiffusionPipeline):
model_output = self.unet(image, timestep).sample
scheduler_output = self.scheduler.step(model_output, timestep, image).prev_sample
result = scheduler_output - scheduler_output + torch.ones_like(scheduler_output)
return scheduler_output
return result