Ravi21 commited on
Commit
ce336c7
·
verified ·
1 Parent(s): 10ac9f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
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)