j-min commited on
Commit
46fd0bc
Β·
1 Parent(s): 66a4e2e

disable cpu offload and xformers

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -52,12 +52,12 @@ class Instance:
52
 
53
  print("Loaded model")
54
 
55
- # This command loads the individual model components on GPU on-demand. So, we don't
56
- # need to explicitly call pipe.to("cuda").
57
- pipe.enable_model_cpu_offload()
58
 
59
- # xformers
60
- pipe.enable_xformers_memory_efficient_attention()
61
  return pipe
62
 
63
  instance = Instance()
 
52
 
53
  print("Loaded model")
54
 
55
+ # # This command loads the individual model components on GPU on-demand. So, we don't
56
+ # # need to explicitly call pipe.to("cuda").
57
+ # pipe.enable_model_cpu_offload()
58
 
59
+ # # xformers
60
+ # pipe.enable_xformers_memory_efficient_attention()
61
  return pipe
62
 
63
  instance = Instance()