Alon77777 commited on
Commit
68f4997
1 Parent(s): 0ea1b17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -34,12 +34,12 @@ 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")
37
- # model_alt = snapshot_download(repo_id="BAAI/AltDiffusion")
38
- # model_alt_m9 = snapshot_download(repo_id="BAAI/AltDiffusion-m9")
39
- # safety_checker = snapshot_download(repo_id="multimodalart/sd-sc")
40
- model_to_load = None # model_alt_m9
41
- # with zipfile.ZipFile("mix.zip", 'r') as zip_ref:
42
- # zip_ref.extractall(".")
43
 
44
  def swap_text(option, base):
45
  resize_width = 768 if base == "v2-768" else 512
 
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")
37
+ model_alt = snapshot_download(repo_id="BAAI/AltDiffusion")
38
+ model_alt_m9 = snapshot_download(repo_id="BAAI/AltDiffusion-m9")
39
+ safety_checker = snapshot_download(repo_id="multimodalart/sd-sc")
40
+ model_to_load = model_alt_m9
41
+ with zipfile.ZipFile("mix.zip", 'r') as zip_ref:
42
+ zip_ref.extractall(".")
43
 
44
  def swap_text(option, base):
45
  resize_width = 768 if base == "v2-768" else 512