How to use pudashi/Atari_small with stable-baselines3:
from huggingface_sb3 import load_from_hub checkpoint = load_from_hub( repo_id="pudashi/Atari_small", filename="{MODEL FILENAME}.zip", )