Yaron Koresh commited on
Commit
0f2dd62
·
verified ·
1 Parent(s): 9085a23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, revision="refs/pr/1", tokenizer="black-forest-labs/FLUX.1-schnell", device=-1, token=os.getenv("hf_token"))
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():