Spaces:

samwell
/

cosmos-predict2-space

Paused

App Files Files Community

cosmos-predict2-space / handler.py

Commit History

Fix: ensure_on_device before each inference to prevent text_encoder drift

d95102e
verified

samwell commited on 13 days ago

Explicitly move all pipeline components to CUDA, use inference_mode

850ecac
verified

samwell commited on 13 days ago

Remove generator entirely to fix CUDA/CPU device mismatch

0c995b9
verified

samwell commited on 13 days ago

Fix generator - remove device spec, use official example pattern

0e4048f
verified

samwell commited on 13 days ago

Fix CUDA device mismatch - resize image and add autocast

2b2bf3c
verified

samwell commited on 13 days ago

Add FastAPI handler

4233b79
verified

samwell commited on 14 days ago