silveroxides commited on
Commit
90963cb
1 Parent(s): 881124a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -62,7 +62,7 @@ def download_file(url, folder_path, filename):
62
  # Download ESRGAN models
63
  download_file("https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealESRGAN_x2.pth?download=true", "models/upscalers/", "RealESRGAN_x2.pth")
64
  download_file("https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealESRGAN_x4.pth?download=true", "models/upscalers/", "RealESRGAN_x4.pth")
65
- # download_file("https://huggingface.co/silveroxides/sdxl_lowstep_beta_v1/resolve/main/pytorch_lora_weights.safetensors?download=true", "models/lora/", "LowStepBeta_v001.safetensors")
66
  # Download the model files
67
  ckpt_dir_realpony = snapshot_download(repo_id="silveroxides/RNS_RealPonyV20")
68
  ckpt_dir_noobai = snapshot_download(repo_id="silveroxides/NoobAI-XL-EPS-1.0-Vwe")
@@ -227,7 +227,7 @@ def generate_image(model_choice, additional_positive_prompt, additional_negative
227
  pipe = pipe_hybridpony
228
 
229
  if use_lowstep_lora:
230
- pipe.unet.load_attn_procs("silveroxides/sdxl_lowstep_beta_v1", weight_name="pytorch_lora_weights.safetensors")
231
 
232
  if use_random_seed:
233
  seed = random.randint(0, 2**32 - 1)
 
62
  # Download ESRGAN models
63
  download_file("https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealESRGAN_x2.pth?download=true", "models/upscalers/", "RealESRGAN_x2.pth")
64
  download_file("https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealESRGAN_x4.pth?download=true", "models/upscalers/", "RealESRGAN_x4.pth")
65
+ download_file("https://huggingface.co/silveroxides/sdxl_lowstep_beta_v1/resolve/main/pytorch_lora_weights.safetensors?download=true", "models/lora/", "LowStepBeta_v001.safetensors")
66
  # Download the model files
67
  ckpt_dir_realpony = snapshot_download(repo_id="silveroxides/RNS_RealPonyV20")
68
  ckpt_dir_noobai = snapshot_download(repo_id="silveroxides/NoobAI-XL-EPS-1.0-Vwe")
 
227
  pipe = pipe_hybridpony
228
 
229
  if use_lowstep_lora:
230
+ pipe.load_lora_weights("./models/lora", weight_name="LowStepBeta_v001.safetensors")
231
 
232
  if use_random_seed:
233
  seed = random.randint(0, 2**32 - 1)