HairStable Bot
commited on
Commit
·
786fe18
1
Parent(s):
a942552
chore: improve error logging for device/cuda flags and traceback
Browse files
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 |
-
|
|
|
|
|
|
|
| 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
|