Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,9 +16,11 @@ def reload_model(new_model_id):
|
|
| 16 |
global pipe, model_id, prev_height, prev_width
|
| 17 |
model_id = new_model_id
|
| 18 |
try:
|
| 19 |
-
|
|
|
|
| 20 |
pipe.reshape(batch_size=1, height=prev_height, width=prev_width, num_images_per_prompt=1)
|
| 21 |
pipe.compile()
|
|
|
|
| 22 |
return f"Model successfully loaded: {model_id}"
|
| 23 |
except Exception as e:
|
| 24 |
return f"Failed to load model: {str(e)}"
|
|
|
|
| 16 |
global pipe, model_id, prev_height, prev_width
|
| 17 |
model_id = new_model_id
|
| 18 |
try:
|
| 19 |
+
print(f"{model_id}...")
|
| 20 |
+
pipe = OVStableDiffusionXLPipeline.from_pretrained(model_id, compile=False)
|
| 21 |
pipe.reshape(batch_size=1, height=prev_height, width=prev_width, num_images_per_prompt=1)
|
| 22 |
pipe.compile()
|
| 23 |
+
print(f"{model_id}!!!")
|
| 24 |
return f"Model successfully loaded: {model_id}"
|
| 25 |
except Exception as e:
|
| 26 |
return f"Failed to load model: {str(e)}"
|