GGPENG commited on
Commit
590639f
·
verified ·
1 Parent(s): c9dcd2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -37,8 +37,12 @@ pipe.enable_attention_slicing()
37
  # state_dict
38
  # )
39
 
 
 
 
40
  pipe.unet.load_attn_procs(
41
- "./pytorch_custom_diffusion_weights.bin"
 
42
  )
43
 
44
  print("Pipeline loaded")
 
37
  # state_dict
38
  # )
39
 
40
+ import os
41
+ print(os.path.getsize("pytorch_custom_diffusion_weights.bin"))
42
+
43
  pipe.unet.load_attn_procs(
44
+ "./pytorch_custom_diffusion_weights.bin",
45
+ weight_name="pytorch_custom_diffusion_weights.bin"
46
  )
47
 
48
  print("Pipeline loaded")