HairStable Bot commited on
Commit
786fe18
·
1 Parent(s): a942552

chore: improve error logging for device/cuda flags and traceback

Browse files
Files changed (1) hide show
  1. Hair_stable_new_fresh/server.py +3 -1
Hair_stable_new_fresh/server.py CHANGED
@@ -177,7 +177,9 @@ def get_hairswap(req: HairSwapRequest, _=Depends(verify_bearer)):
177
  )
178
  LOGGER.info("Hair transfer completed successfully")
179
  except Exception as e:
180
- LOGGER.error(f"Hair transfer failed: {str(e)}")
 
 
181
  raise HTTPException(status_code=500, detail=f"Hair transfer failed: {str(e)}")
182
 
183
  # Save result
 
177
  )
178
  LOGGER.info("Hair transfer completed successfully")
179
  except Exception as e:
180
+ import traceback
181
+ tb = traceback.format_exc()
182
+ LOGGER.error(f"Hair transfer failed: {str(e)} | device={model.device if hasattr(model, 'device') else 'n/a'} cuda_available={torch.cuda.is_available()}\n{tb}")
183
  raise HTTPException(status_code=500, detail=f"Hair transfer failed: {str(e)}")
184
 
185
  # Save result