LanHarmony commited on
Commit
db07143
·
1 Parent(s): a05a4de

introduce control net

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -128,7 +128,7 @@ class ConversationBot:
128
  # self.image2depth = image2depth()
129
  # self.depth2image = depth2image(device="cuda:7")
130
  # self.image2normal = image2normal()
131
- self.normal2image = normal2image(device="cuda:5")
132
  self.pix2pix = Pix2Pix(device="cuda:3")
133
  self.memory = ConversationBufferMemory(memory_key="chat_history", output_key='output')
134
  self.tools = [
 
128
  # self.image2depth = image2depth()
129
  # self.depth2image = depth2image(device="cuda:7")
130
  # self.image2normal = image2normal()
131
+ # self.normal2image = normal2image(device="cuda:5")
132
  self.pix2pix = Pix2Pix(device="cuda:3")
133
  self.memory = ConversationBufferMemory(memory_key="chat_history", output_key='output')
134
  self.tools = [