Spaces:
Runtime error
Runtime error
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -209,7 +209,7 @@ class FluxWithCFGPipeline(FluxPipeline):
|
|
| 209 |
|
| 210 |
class FluxWithCFGPipeline(FluxPipeline):
|
| 211 |
@torch.inference_mode()
|
| 212 |
-
def
|
| 213 |
self,
|
| 214 |
prompt: Union[str, List[str]] = None,
|
| 215 |
prompt_2: Optional[Union[str, List[str]]] = None,
|
|
|
|
| 209 |
|
| 210 |
class FluxWithCFGPipeline(FluxPipeline):
|
| 211 |
@torch.inference_mode()
|
| 212 |
+
def generate_image(
|
| 213 |
self,
|
| 214 |
prompt: Union[str, List[str]] = None,
|
| 215 |
prompt_2: Optional[Union[str, List[str]]] = None,
|