samwell
/

cosmos-predict2-space

Model card Files Files and versions

cosmos-predict2-space / handler.py

Commit History

Fix CUDA device mismatch - resize image and add autocast

5cf4cc1
verified

samwell commited on 10 days ago