fffiloni commited on
Commit
4a902da
·
verified ·
1 Parent(s): 2f22a68

set cache dir

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -8,6 +8,7 @@ def generate_image(prompt):
8
  args.task = "single"
9
  args.prompt = prompt
10
  args.model = "sd-turbo" # or another supported model
 
11
  args.save_dir = "./outputs"
12
  args.save_all_images = True
13
 
 
8
  args.task = "single"
9
  args.prompt = prompt
10
  args.model = "sd-turbo" # or another supported model
11
+ args.cache_dir = "./HF_model_cache"
12
  args.save_dir = "./outputs"
13
  args.save_all_images = True
14