Alexander McKinney commited on
Commit
a833338
1 Parent(s): d71e882

adds token reading

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -13,6 +13,8 @@ from transformers.models.detr.feature_extraction_detr import rgb_to_id
13
 
14
  from diffusers import StableDiffusionInpaintPipeline
15
 
 
 
16
  torch.inference_mode()
17
  torch.no_grad()
18
 
@@ -29,7 +31,8 @@ def load_diffusion_pipeline(model_name: str = 'runwayml/stable-diffusion-inpaint
29
  return StableDiffusionInpaintPipeline.from_pretrained(
30
  model_name,
31
  revision='fp16',
32
- torch_dtype=torch.float16
 
33
  )
34
 
35
  # Device helper
 
13
 
14
  from diffusers import StableDiffusionInpaintPipeline
15
 
16
+ auth_token = os.environ.get("READ_TOKEN")
17
+
18
  torch.inference_mode()
19
  torch.no_grad()
20
 
 
31
  return StableDiffusionInpaintPipeline.from_pretrained(
32
  model_name,
33
  revision='fp16',
34
+ torch_dtype=torch.float16,
35
+ use_auth_token=auth_token
36
  )
37
 
38
  # Device helper