Astridkraft commited on
Commit
d57291a
·
verified ·
1 Parent(s): 2f0e9d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -119,15 +119,15 @@ class TextToImageProgressCallback:
119
 
120
  class ImageToImageProgressCallback:
121
  def __init__(self, progress, total_steps, strength):
122
- self.progress = progress
123
- self.total_steps = total_steps
124
- self.current_step = 0
125
- self.strength = strength
126
- self.actual_total_steps = None
127
 
128
 
129
  def __call__(self, step, timestep, latents):
130
- self.current_step = step + 1
131
 
132
  # Korrekte Berechnung der tatsächlichen Steps
133
  if self.actual_total_steps is None:
 
119
 
120
  class ImageToImageProgressCallback:
121
  def __init__(self, progress, total_steps, strength):
122
+ self.progress = progress
123
+ self.total_steps = total_steps
124
+ self.current_step = 0
125
+ self.strength = strength
126
+ self.actual_total_steps = None
127
 
128
 
129
  def __call__(self, step, timestep, latents):
130
+ self.current_step = step + 1
131
 
132
  # Korrekte Berechnung der tatsächlichen Steps
133
  if self.actual_total_steps is None: