Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ PIPE = None
|
|
41 |
def pipe_t2i():
|
42 |
global PIPE
|
43 |
if PIPE is None:
|
44 |
-
PIPE = pipeline("text-to-image", model="black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16,
|
45 |
return PIPE
|
46 |
|
47 |
def pipe_i2i():
|
|
|
41 |
def pipe_t2i():
|
42 |
global PIPE
|
43 |
if PIPE is None:
|
44 |
+
PIPE = pipeline("text-to-image", model="black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, tokenizer="black-forest-labs/FLUX.1-schnell", device=-1, token=os.getenv("hf_token"))
|
45 |
return PIPE
|
46 |
|
47 |
def pipe_i2i():
|