Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -427,6 +427,16 @@ class OVStableDiffusionPipeline(DiffusionPipeline):
|
|
| 427 |
|
| 428 |
core=pv.Core()
|
| 429 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 430 |
text_enc=core.compile_model(text_encoder_path,device.value)
|
| 431 |
|
| 432 |
unet_model=core.compile_model(unet_path,device.value)
|
|
|
|
| 427 |
|
| 428 |
core=pv.Core()
|
| 429 |
|
| 430 |
+
import ipywidgets as widgets
|
| 431 |
+
device=widgets.Dropdown(
|
| 432 |
+
options=core.available_devices+["AUTO"],
|
| 433 |
+
value="CPU",
|
| 434 |
+
desciption="Device:",
|
| 435 |
+
disabled=False,
|
| 436 |
+
)
|
| 437 |
+
device
|
| 438 |
+
|
| 439 |
+
|
| 440 |
text_enc=core.compile_model(text_encoder_path,device.value)
|
| 441 |
|
| 442 |
unet_model=core.compile_model(unet_path,device.value)
|