Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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}!!!")
|