Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
fcfc6b2
1
Parent(s):
7befb0b
Update custom_pipeline.py
Browse files- custom_pipeline.py +1 -1
custom_pipeline.py
CHANGED
|
@@ -58,7 +58,7 @@ def prepare_timesteps(
|
|
| 58 |
class FluxWithCFGPipeline(FluxPipeline):
|
| 59 |
|
| 60 |
@torch.inference_mode()
|
| 61 |
-
def
|
| 62 |
self,
|
| 63 |
prompt: Union[str, List[str]] = None,
|
| 64 |
prompt_2: Optional[Union[str, List[str]]] = None,
|
|
|
|
| 58 |
class FluxWithCFGPipeline(FluxPipeline):
|
| 59 |
|
| 60 |
@torch.inference_mode()
|
| 61 |
+
def generate_image(
|
| 62 |
self,
|
| 63 |
prompt: Union[str, List[str]] = None,
|
| 64 |
prompt_2: Optional[Union[str, List[str]]] = None,
|