KarthikAI commited on
Commit
12c11e9
·
verified ·
1 Parent(s): 8ecbb74

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -0
utils.py CHANGED
@@ -2,6 +2,7 @@ import os
2
  os.environ["HF_HOME"] = "/data/huggingface"
3
  os.environ["TRANSFORMERS_CACHE"] = "/data/huggingface"
4
  os.makedirs("/data/huggingface/hub", exist_ok=True)
 
5
 
6
  import torch
7
  from diffusers import StableDiffusionImg2ImgPipeline
 
2
  os.environ["HF_HOME"] = "/data/huggingface"
3
  os.environ["TRANSFORMERS_CACHE"] = "/data/huggingface"
4
  os.makedirs("/data/huggingface/hub", exist_ok=True)
5
+ os.makedirs("/data/huggingface/clip_vision_model", exist_ok=True)
6
 
7
  import torch
8
  from diffusers import StableDiffusionImg2ImgPipeline