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:
|