Open-Sora
Collection
A Series of Open-Sora Models β’ 11 items β’ Updated β’ 10
How to use hpcai-tech/OpenSora-STDiT-v2-stage2 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("feature-extraction", model="hpcai-tech/OpenSora-STDiT-v2-stage2", trust_remote_code=True) # Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("hpcai-tech/OpenSora-STDiT-v2-stage2", trust_remote_code=True, dtype="auto")