macrdel commited on
Commit
a005ea3
·
1 Parent(s): 72437e3

update app.py

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -0
app.py CHANGED
@@ -14,7 +14,7 @@ model_repo = "black-forest-labs/FLUX.1-dev" # Replace to the model you would li
14
  pipe = FluxPipeline.from_pretrained(model_repo,
15
  torch_dtype=torch_dtype,
16
  token=f"{os.environ.get('tkn')}",
17
- force_download=True,
18
  ).to(device)
19
  #pipe.enable_model_cpu_offload()
20
 
 
14
  pipe = FluxPipeline.from_pretrained(model_repo,
15
  torch_dtype=torch_dtype,
16
  token=f"{os.environ.get('tkn')}",
17
+ #force_download=True,
18
  ).to(device)
19
  #pipe.enable_model_cpu_offload()
20
 
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  diffusers
2
  transformers
3
  torch
 
1
+ accelerate
2
  diffusers
3
  transformers
4
  torch