Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
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,
|