jbilcke-hf HF staff commited on
Commit
472ac56
·
1 Parent(s): 3b1d138

Update app.py

Browse files

doing some experiments

Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -46,6 +46,8 @@ if torch.cuda.is_available():
46
  pipe.unet = torch.compile(pipe.unet,
47
  mode='reduce-overhead',
48
  fullgraph=True)
 
 
49
  else:
50
  pipe = None
51
  refiner = None
 
46
  pipe.unet = torch.compile(pipe.unet,
47
  mode='reduce-overhead',
48
  fullgraph=True)
49
+
50
+ pipeline.unet.load_attn_procs('artificialguybr/360Redmond')
51
  else:
52
  pipe = None
53
  refiner = None