Your Name
commited on
Commit
·
d6435d5
1
Parent(s):
db91d32
- src/pipeline.py +2 -1
src/pipeline.py
CHANGED
|
@@ -18,6 +18,7 @@ Pipeline = None
|
|
| 18 |
|
| 19 |
ckpt_id = "black-forest-labs/FLUX.1-schnell"
|
| 20 |
|
|
|
|
| 21 |
def empty_cache():
|
| 22 |
start = time.time()
|
| 23 |
gc.collect()
|
|
@@ -63,5 +64,5 @@ def infer(request: TextToImageRequest, pipeline: Pipeline) -> Image:
|
|
| 63 |
image = create_gray_image(request.width, request.height)
|
| 64 |
pass
|
| 65 |
|
| 66 |
-
|
| 67 |
return(image)
|
|
|
|
| 18 |
|
| 19 |
ckpt_id = "black-forest-labs/FLUX.1-schnell"
|
| 20 |
|
| 21 |
+
|
| 22 |
def empty_cache():
|
| 23 |
start = time.time()
|
| 24 |
gc.collect()
|
|
|
|
| 64 |
image = create_gray_image(request.width, request.height)
|
| 65 |
pass
|
| 66 |
|
| 67 |
+
|
| 68 |
return(image)
|