huggingface / huggingface/diffusers

pipe.scheduler.sigmas change after forward call

Open
#7,904 6 comments 0 reactions 0 assignees View on GitHub
bug stale
Dominant language
Python
Stars
34.5k
Forks
7.3k
Avg merge
3d 3h
Merged PRs (30d)
91

Description

### Describe the bug

I noticed this issue when trying to manually modify the sigmas in my pipeline scheduler. Despite changing the values in this tensor, the sigmas tensor shrinks to the number of inference steps after one inference instance. Subsequent calls no longer change the values of the sigmas.

### Reproduction

pipe.scheduler.sigmas = pipe.scheduler.sigmas * 4 # manually modifying sigmas
_ = pipe(..., num_inference_steps = N)
print(pipe.scheduler.sigmas) # len is N + 1, values do not reflect the manual modifications and no longer change in subsequent calls

### Logs

_No response_

### System Info

diffusers==0.27.2

### Who can help?

_No response_

Contributor guide

Open the contributing guide

Research direction

Start at the scheduler entry point used by pipe(...) and trace how scheduler.sigmas is initialized or resized during an inference call. Reproduce the example with diffusers 0.27.2, then verify that manually modified sigmas retain the intended behavior across the first and subsequent calls.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Bug
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.