mfranzon commited on
Commit
4fd608e
1 Parent(s): e5fc540

fix utils, add token

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -2,7 +2,7 @@ import torch
2
  from diffusers import StableDiffusionImg2ImgPipeline
3
 
4
  import os
5
- TOKEN = os.environ.get('sd') or True
6
 
7
  torch.cuda.empty_cache()
8
  device = "cpu"
 
2
  from diffusers import StableDiffusionImg2ImgPipeline
3
 
4
  import os
5
+ TOKEN = os.environ.get('TOKEN_SD') or True
6
 
7
  torch.cuda.empty_cache()
8
  device = "cpu"