michaelapplydesign commited on
Commit
92ac539
1 Parent(s): 28471f6
Files changed (1) hide show
  1. pipelines.py +1 -1
pipelines.py CHANGED
@@ -74,7 +74,7 @@ class SDPipeline:
74
  )
75
 
76
  self.pipe.enable_xformers_memory_efficient_attention()
77
- # self.pipe = self.pipe.to("cuda")
78
 
79
  self.waiting_queue = []
80
  self.count = 0
 
74
  )
75
 
76
  self.pipe.enable_xformers_memory_efficient_attention()
77
+ self.pipe = self.pipe.to("cuda")
78
 
79
  self.waiting_queue = []
80
  self.count = 0