treasuraid commited on
Commit
fc3e1a1
·
1 Parent(s): dc6157e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -25,7 +25,8 @@ def load_pipeline_w_lora() :
25
 
26
  # Load LoRa attn layer weights to unet attn layers
27
  unet.load_attn_procs(LoRa_DIR)
28
-
 
29
  # Load pipeline
30
  pipeline = DiffusionPipeline.from_pretrained(
31
  MODEL_REPO,
 
25
 
26
  # Load LoRa attn layer weights to unet attn layers
27
  unet.load_attn_procs(LoRa_DIR)
28
+ print('loaded')
29
+
30
  # Load pipeline
31
  pipeline = DiffusionPipeline.from_pretrained(
32
  MODEL_REPO,