Update app.py
Browse files
app.py
CHANGED
|
@@ -342,7 +342,7 @@ def get_mrt():
|
|
| 342 |
guidance_weight=5.0,
|
| 343 |
device="gpu",
|
| 344 |
checkpoint_dir=ckpt_dir,
|
| 345 |
-
lazy=
|
| 346 |
)
|
| 347 |
# If no assets loaded yet, and a repo is configured, load them now.
|
| 348 |
if asset_manager.mean_embed is None and asset_manager.centroids is None:
|
|
|
|
| 342 |
guidance_weight=5.0,
|
| 343 |
device="gpu",
|
| 344 |
checkpoint_dir=ckpt_dir,
|
| 345 |
+
lazy=True # Avoid warm start that causes XlaCallModuleOp version error
|
| 346 |
)
|
| 347 |
# If no assets loaded yet, and a repo is configured, load them now.
|
| 348 |
if asset_manager.mean_embed is None and asset_manager.centroids is None:
|