Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
310a7f1
1
Parent(s):
889a4b9
add torchvision to requirement
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ from src.attention_processor import FluxBlendedAttnProcessor2_0
|
|
| 11 |
from src.utils_sample import set_seed, resize_and_add_margin
|
| 12 |
import os
|
| 13 |
|
| 14 |
-
|
| 15 |
|
| 16 |
@spaces.GPU
|
| 17 |
def process_image_and_text(image, scale, seed, text, pipe):
|
|
@@ -140,7 +140,6 @@ def create_app():
|
|
| 140 |
dtype = torch.bfloat16
|
| 141 |
token = os.environ.get("HF_TOKEN")
|
| 142 |
|
| 143 |
-
pipe = None
|
| 144 |
pipe = FluxPipeline.from_pretrained(
|
| 145 |
"black-forest-labs/FLUX.1-dev",
|
| 146 |
torch_dtype=dtype,
|
|
|
|
| 11 |
from src.utils_sample import set_seed, resize_and_add_margin
|
| 12 |
import os
|
| 13 |
|
| 14 |
+
pipe = None
|
| 15 |
|
| 16 |
@spaces.GPU
|
| 17 |
def process_image_and_text(image, scale, seed, text, pipe):
|
|
|
|
| 140 |
dtype = torch.bfloat16
|
| 141 |
token = os.environ.get("HF_TOKEN")
|
| 142 |
|
|
|
|
| 143 |
pipe = FluxPipeline.from_pretrained(
|
| 144 |
"black-forest-labs/FLUX.1-dev",
|
| 145 |
torch_dtype=dtype,
|