Spaces:
vilarin
/
Running on Zero

vilarin commited on
Commit
14ce716
1 Parent(s): 9f5b615

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -64,6 +64,7 @@ def enable_lora(lora_scale,lora_add):
64
  url = f'https://huggingface.co/{lora_add}/tree/main'
65
  lora_name = scrape_lora_link(url)
66
  if lora_name:
 
67
  pipe.load_lora_weights(lora_add, weight_name=lora_name)
68
  pipe.fuse_lora(lora_scale=lora_scale)
69
  gr.Info(f"{lora_add} Loaded")
@@ -71,6 +72,7 @@ def enable_lora(lora_scale,lora_add):
71
  try:
72
  pipe.load_lora_weights(lora_add)
73
  pipe.fuse_lora(lora_scale=lora_scale)
 
74
  except:
75
  raise gr.Error(f"{lora_add} Load fail, check again.")
76
 
 
64
  url = f'https://huggingface.co/{lora_add}/tree/main'
65
  lora_name = scrape_lora_link(url)
66
  if lora_name:
67
+ print(f'lora loading: {lora_add}/{lora_name}')
68
  pipe.load_lora_weights(lora_add, weight_name=lora_name)
69
  pipe.fuse_lora(lora_scale=lora_scale)
70
  gr.Info(f"{lora_add} Loaded")
 
72
  try:
73
  pipe.load_lora_weights(lora_add)
74
  pipe.fuse_lora(lora_scale=lora_scale)
75
+ gr.Info(f"{lora_add} Loaded")
76
  except:
77
  raise gr.Error(f"{lora_add} Load fail, check again.")
78