multimodalart HF staff commited on
Commit
1b5780b
1 Parent(s): 5798e9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,7 +30,7 @@ css = '''
30
  maximum_concepts = 3
31
 
32
  #Pre download the files
33
- if(not is_shared_ui and not is_gpu_associated):
34
  model_v1 = snapshot_download(repo_id="multimodalart/sd-fine-tunable")
35
  model_v2 = snapshot_download(repo_id="stabilityai/stable-diffusion-2")
36
  model_v2_512 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-base")
@@ -58,7 +58,7 @@ def swap_text(option, base):
58
  return [f"You are going to train a `style`, upload 10-20 images of the style you are planning on training on. You can use services like <a style='text-decoration: underline' target='_blank' href='https://www.birme.net/?target_width={resize_width}&target_height={resize_width}'>birme</a> for smart cropping. Name the files with the words you would like {mandatory_liability}:", '''<img src="file/trsl_style.png" />''', f"You should name your concept with a unique made up word that has low chance of the model already knowing it (e.g.: `{instance_prompt_example}` here). Images will be automatically cropped to {resize_width}x{resize_width}", freeze_for, gr.update(visible=False)]
59
 
60
  def swap_base_model(selected_model):
61
- if(not is_shared_ui and not is_gpu_associated):
62
  global model_to_load
63
  if(selected_model == "v1-5"):
64
  model_to_load = model_v1
 
30
  maximum_concepts = 3
31
 
32
  #Pre download the files
33
+ if(is_gpu_associated):
34
  model_v1 = snapshot_download(repo_id="multimodalart/sd-fine-tunable")
35
  model_v2 = snapshot_download(repo_id="stabilityai/stable-diffusion-2")
36
  model_v2_512 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-base")
 
58
  return [f"You are going to train a `style`, upload 10-20 images of the style you are planning on training on. You can use services like <a style='text-decoration: underline' target='_blank' href='https://www.birme.net/?target_width={resize_width}&target_height={resize_width}'>birme</a> for smart cropping. Name the files with the words you would like {mandatory_liability}:", '''<img src="file/trsl_style.png" />''', f"You should name your concept with a unique made up word that has low chance of the model already knowing it (e.g.: `{instance_prompt_example}` here). Images will be automatically cropped to {resize_width}x{resize_width}", freeze_for, gr.update(visible=False)]
59
 
60
  def swap_base_model(selected_model):
61
+ if(is_gpu_associated):
62
  global model_to_load
63
  if(selected_model == "v1-5"):
64
  model_to_load = model_v1