adityas129 commited on
Commit
c7eb087
·
verified ·
1 Parent(s): e1e2507

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=False
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: