Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -119,15 +119,15 @@ class TextToImageProgressCallback:
|
|
| 119 |
|
| 120 |
class ImageToImageProgressCallback:
|
| 121 |
def __init__(self, progress, total_steps, strength):
|
| 122 |
-
|
| 123 |
-
|
| 124 |
-
|
| 125 |
-
|
| 126 |
-
|
| 127 |
|
| 128 |
|
| 129 |
def __call__(self, step, timestep, latents):
|
| 130 |
-
|
| 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:
|