chaojiemao commited on
Commit
d9a649e
·
verified ·
1 Parent(s): d82aa15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -76,6 +76,14 @@ class ChatBotUI(object):
76
  if model_cfg.IS_DEFAULT: self.default_model_name = model_name
77
  self.model_choices[model_name] = model_cfg
78
  print('Models: ', self.model_choices.keys())
 
 
 
 
 
 
 
 
79
  assert len(self.model_choices) > 0
80
  if self.default_model_name == "": self.default_model_name = self.model_choices.keys()[0]
81
  self.model_name = self.default_model_name
 
76
  if model_cfg.IS_DEFAULT: self.default_model_name = model_name
77
  self.model_choices[model_name] = model_cfg
78
  print('Models: ', self.model_choices.keys())
79
+
80
+ FS.get_object_to_local_file("ms://AI-ModelScope/FLUX.1-dev@flux1-dev.safetensors")
81
+ FS.get_object_to_local_file("ms://AI-ModelScope/FLUX.1-dev@ae.safetensors")
82
+ FS.get_dir_to_local_dir("ms://AI-ModelScope/FLUX.1-dev@text_encoder_2/")
83
+ FS.get_dir_to_local_dir("ms://AI-ModelScope/FLUX.1-dev@tokenizer_2/")
84
+ FS.get_dir_to_local_dir("ms://AI-ModelScope/FLUX.1-dev@text_encoder/")
85
+ FS.get_dir_to_local_dir("ms://AI-ModelScope/FLUX.1-dev@tokenizer/")
86
+
87
  assert len(self.model_choices) > 0
88
  if self.default_model_name == "": self.default_model_name = self.model_choices.keys()[0]
89
  self.model_name = self.default_model_name