Spaces:
Runtime error
Runtime error
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 = [
|