MeYourHint commited on
Commit
a25d22e
·
1 Parent(s): 62d8243
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -116,7 +116,7 @@ def generate_component(generate_function, text):
116
 
117
 
118
  if not os.path.exists("checkpoints/t2m"):
119
- os.system("bash prepare/download_models.sh")
120
 
121
 
122
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
116
 
117
 
118
  if not os.path.exists("checkpoints/t2m"):
119
+ os.system("bash prepare/download_models_demo.sh")
120
 
121
 
122
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")