RamAnanth1 commited on
Commit
7beb833
1 Parent(s): a4d0cea

Download the missing CLIP folder

Browse files
Files changed (1) hide show
  1. model.py +3 -0
model.py CHANGED
@@ -71,10 +71,13 @@ class Model:
71
  base_model_file = "https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt"
72
  sketch_adapter_file = "https://huggingface.co/TencentARC/T2I-Adapter/resolve/main/models/t2iadapter_sketch_sd14v1.pth"
73
  pidinet_file = model_path+"table5_pidinet.pth"
 
74
 
75
  subprocess.run(shlex.split(f'wget {base_model_file} -O models/sd-v1-4.ckpt'))
76
  subprocess.run(shlex.split(f'wget {sketch_adapter_file} -O models/t2iadapter_sketch_sd14v1.pth'))
77
  subprocess.run(shlex.split(f'wget {pidinet_file} -O models/table5_pidinet.pth'))
 
 
78
 
79
  model = load_model_from_config(config, "models/sd-v1-4.ckpt").to(device)
80
  current_base = 'sd-v1-4.ckpt'
 
71
  base_model_file = "https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt"
72
  sketch_adapter_file = "https://huggingface.co/TencentARC/T2I-Adapter/resolve/main/models/t2iadapter_sketch_sd14v1.pth"
73
  pidinet_file = model_path+"table5_pidinet.pth"
74
+ clip_file = "https://huggingface.co/openai/clip-vit-large-patch14/resolve/main/*"
75
 
76
  subprocess.run(shlex.split(f'wget {base_model_file} -O models/sd-v1-4.ckpt'))
77
  subprocess.run(shlex.split(f'wget {sketch_adapter_file} -O models/t2iadapter_sketch_sd14v1.pth'))
78
  subprocess.run(shlex.split(f'wget {pidinet_file} -O models/table5_pidinet.pth'))
79
+ subprocess.run(shlex.split(f'wget {clip_file} --mirror --no-parent -O models/clip-vit-large-patch14'))
80
+
81
 
82
  model = load_model_from_config(config, "models/sd-v1-4.ckpt").to(device)
83
  current_base = 'sd-v1-4.ckpt'