Spaces:
Runtime error
Runtime error
LanHarmony
commited on
Commit
·
db07143
1
Parent(s):
a05a4de
introduce control net
Browse files
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 = [
|