noahfaire commited on
Commit
55b4118
·
1 Parent(s): ebe47d5

add accelerate

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -43,7 +43,7 @@ pipe = StableDiffusionInpaintPipeline.from_pretrained(
43
  model_path,
44
  revision="fp16",
45
  torch_dtype=torch.float16,
46
- use_auth_token=False
47
  ).to(device)
48
 
49
  img_url = "https://cdn.faire.com/fastly/893b071985d70819da5f0d485f1b1bb97ee4f16a6e14ef1bdd4a086b3588be58.png" # wino
 
43
  model_path,
44
  revision="fp16",
45
  torch_dtype=torch.float16,
46
+ use_auth_token=True
47
  ).to(device)
48
 
49
  img_url = "https://cdn.faire.com/fastly/893b071985d70819da5f0d485f1b1bb97ee4f16a6e14ef1bdd4a086b3588be58.png" # wino
requirements.txt CHANGED
@@ -5,4 +5,5 @@ transformers
5
  typing
6
  diffusers
7
  git+https://github.com/huggingface/diffusers.git
8
- huggingface_hub
 
 
5
  typing
6
  diffusers
7
  git+https://github.com/huggingface/diffusers.git
8
+ huggingface_hub
9
+ accelerate