None1145 commited on
Commit
c5d4737
·
verified ·
1 Parent(s): a453abb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def reload_model(new_model_id):
18
  try:
19
  print(f"{model_id}...")
20
  pipe = OVStableDiffusionXLPipeline.from_pretrained(model_id, compile=False)
21
- pipe.to("gpu")
22
  pipe.reshape(batch_size=1, height=prev_height, width=prev_width, num_images_per_prompt=1)
23
  pipe.compile()
24
  print(f"{model_id}!!!")
 
18
  try:
19
  print(f"{model_id}...")
20
  pipe = OVStableDiffusionXLPipeline.from_pretrained(model_id, compile=False)
21
+ # pipe.to("gpu")
22
  pipe.reshape(batch_size=1, height=prev_height, width=prev_width, num_images_per_prompt=1)
23
  pipe.compile()
24
  print(f"{model_id}!!!")