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)
|