Spaces:
Running
on
Zero
Running
on
Zero
nick_93
commited on
Commit
·
a920397
1
Parent(s):
caddeb4
init
Browse files
app.py
CHANGED
|
@@ -363,7 +363,9 @@ def create_demo(model):
|
|
| 363 |
|
| 364 |
def main():
|
| 365 |
model = ControlNetDepthDesignModelMulti()
|
|
|
|
| 366 |
print('Models uploaded successfully')
|
|
|
|
| 367 |
|
| 368 |
title = "# StableDesign"
|
| 369 |
description = """
|
|
|
|
| 363 |
|
| 364 |
def main():
|
| 365 |
model = ControlNetDepthDesignModelMulti()
|
| 366 |
+
model = model.to(device)
|
| 367 |
print('Models uploaded successfully')
|
| 368 |
+
print(model.device)
|
| 369 |
|
| 370 |
title = "# StableDesign"
|
| 371 |
description = """
|