43ntropy
/

NEvo / examples /override_models.py
43ntropy's picture
Duplicate from epfl-neuroai/NEvo
1e2bb2f
Raw
History Blame Contribute Delete
478 Bytes
from stimulus_synthesis import NevoPipeline
pipe = NevoPipeline.from_pretrained(
"epfl-neuroai/NEvo",
synthesis_config={
"text_to_image_model_id": "stabilityai/sdxl-turbo",
"image_to_video_model_id": "Lightricks/LTX-Video-0.9.8-13B-distilled",
"encoder_model_id": "epfl-neuroai/vjepa2-encoder-basic",
},
)
out = pipe(target={"type": "indices", "indices": [0, 1, 2]}, seed_prompts=["a moving person"])
print(out.best_prompt, out.best_score)