Spaces:
mashroo
/
Runtime error

YoussefAnso's picture
Update device handling in TwoStagePipeline to default to 'cuda' and adjust model loading to use 'cpu' for compatibility. Ensure consistent device usage across stages and improve code clarity.
7391a72