Spaces:
Running
on
Zero
Running
on
Zero
Prime Cai
commited on
Commit
·
c22dc8b
1
Parent(s):
bf9c94f
fix gemini prompt
Browse files- pipeline.py +2 -1
pipeline.py
CHANGED
|
@@ -722,7 +722,8 @@ class FluxConditionalPipeline(DiffusionPipeline, SD3LoraLoaderMixin):
|
|
| 722 |
|
| 723 |
device = self._execution_device
|
| 724 |
|
| 725 |
-
|
|
|
|
| 726 |
# if gemini_prompt:
|
| 727 |
# while True:
|
| 728 |
# try:
|
|
|
|
| 722 |
|
| 723 |
device = self._execution_device
|
| 724 |
|
| 725 |
+
if gemini_prompt:
|
| 726 |
+
prompt = enhance_prompt(image, prompt)
|
| 727 |
# if gemini_prompt:
|
| 728 |
# while True:
|
| 729 |
# try:
|