├── D4RL ├── .github │ ├── FUNDING.yml │ ├── ISSUE_TEMPLATE │ │ ├── bug.md │ │ ├── proposal.md │ │ └── question.md │ ├── PULL_REQUEST_TEMPLATE.md │ └── stale.yml ├── .gitignore ├── CODE_OF_CONDUCT.rst ├── LICENSE ├── MANIFEST.in ├── README.md ├── d4rl │ ├── __init__.py │ ├── carla │ │ ├── __init__.py │ │ ├── carla_env.py │ │ ├── data_collection_agent_lane.py │ │ ├── data_collection_town.py │ │ └── town_agent.py │ ├── flow │ │ ├── __init__.py │ │ ├── bottleneck.py │ │ ├── merge.py │ │ └── traffic_light_grid.py │ ├── gym_bullet │ │ ├── __init__.py │ │ └── gym_envs.py │ ├── gym_minigrid │ │ ├── __init__.py │ │ ├── envs │ │ │ ├── __init__.py │ │ │ ├── empty.py │ │ │ └── fourrooms.py │ │ ├── fourroom_controller.py │ │ ├── minigrid.py │ │ ├── register.py │ │ ├── rendering.py │ │ ├── roomgrid.py │ │ ├── window.py │ │ └── wrappers.py │ ├── gym_mujoco │ │ ├── __init__.py │ │ └── gym_envs.py │ ├── hand_manipulation_suite │ │ ├── Adroit │ │ │ ├── .gitignore │ │ │ ├── Adroit_hand.xml │ │ │ ├── Adroit_hand_withOverlay.xml │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── gallery │ │ │ │ ├── news.JPG │ │ │ │ └── projects.JPG │ │ │ └── resources │ │ │ │ ├── assets.xml │ │ │ │ ├── chain.xml │ │ │ │ ├── chain1.xml │ │ │ │ ├── joint_position_actuation.xml │ │ │ │ ├── meshes │ │ │ │ ├── F1.stl │ │ │ │ ├── F2.stl │ │ │ │ ├── F3.stl │ │ │ │ ├── TH1_z.stl │ │ │ │ ├── TH2_z.stl │ │ │ │ ├── TH3_z.stl │ │ │ │ ├── arm_base.stl │ │ │ │ ├── arm_trunk.stl │ │ │ │ ├── arm_trunk_asmbly.stl │ │ │ │ ├── distal_ellipsoid.stl │ │ │ │ ├── elbow_flex.stl │ │ │ │ ├── elbow_rotate_motor.stl │ │ │ │ ├── elbow_rotate_muscle.stl │ │ │ │ ├── forearm_Cy_PlateAsmbly(muscle_cone).stl │ │ │ │ ├── forearm_Cy_PlateAsmbly.stl │ │ │ │ ├── forearm_PlateAsmbly.stl │ │ │ │ ├── forearm_electric.stl │ │ │ │ ├── forearm_electric_cvx.stl │ │ │ │ ├── forearm_muscle.stl │ │ │ │ ├── forearm_simple.stl │ │ │ │ ├── forearm_simple_cvx.stl │ │ │ │ ├── forearm_weight.stl │ │ │ │ ├── knuckle.stl │ │ │ │ ├── lfmetacarpal.stl │ │ │ │ ├── palm.stl │ │ │ │ ├── upper_arm.stl │ │ │ │ ├── upper_arm_asmbl_shoulder.stl │ │ │ │ ├── upper_arm_ass.stl │ │ │ │ └── wrist.stl │ │ │ │ └── tendon_torque_actuation.xml │ │ ├── __init__.py │ │ ├── assets │ │ │ ├── DAPG_Adroit.xml │ │ │ ├── DAPG_assets.xml │ │ │ ├── DAPG_door.xml │ │ │ ├── DAPG_hammer.xml │ │ │ ├── DAPG_pen.xml │ │ │ ├── DAPG_relocate.xml │ │ │ └── tasks.jpg │ │ ├── door_v0.py │ │ ├── hammer_v0.py │ │ ├── pen_v0.py │ │ └── relocate_v0.py │ ├── infos.py │ ├── kitchen │ │ ├── __init__.py │ │ ├── adept_envs │ │ │ ├── .pylintrc │ │ │ ├── .style.yapf │ │ │ ├── __init__.py │ │ │ ├── base_robot.py │ │ │ ├── franka │ │ │ │ ├── __init__.py │ │ │ │ ├── assets │ │ │ │ │ └── franka_kitchen_jntpos_act_ab.xml │ │ │ │ ├── kitchen_multitask_v0.py │ │ │ │ └── robot │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── franka_config.xml │ │ │ │ │ └── franka_robot.py │ │ │ ├── mujoco_env.py │ │ │ ├── robot_env.py │ │ │ ├── simulation │ │ │ │ ├── __init__.py │ │ │ │ ├── module.py │ │ │ │ ├── renderer.py │ │ │ │ └── sim_robot.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── config.py │ │ │ │ ├── configurable.py │ │ │ │ ├── constants.py │ │ │ │ ├── parse_demos.py │ │ │ │ └── quatmath.py │ │ ├── adept_models │ │ │ ├── .gitignore │ │ │ ├── CONTRIBUTING.public.md │ │ │ ├── LICENSE │ │ │ ├── README.public.md │ │ │ ├── __init__.py │ │ │ ├── kitchen │ │ │ │ ├── assets │ │ │ │ │ ├── backwall_asset.xml │ │ │ │ │ ├── backwall_chain.xml │ │ │ │ │ ├── counters_asset.xml │ │ │ │ │ ├── counters_chain.xml │ │ │ │ │ ├── hingecabinet_asset.xml │ │ │ │ │ ├── hingecabinet_chain.xml │ │ │ │ │ ├── kettle_asset.xml │ │ │ │ │ ├── kettle_chain.xml │ │ │ │ │ ├── microwave_asset.xml │ │ │ │ │ ├── microwave_chain.xml │ │ │ │ │ ├── oven_asset.xml │ │ │ │ │ ├── oven_chain.xml │ │ │ │ │ ├── slidecabinet_asset.xml │ │ │ │ │ └── slidecabinet_chain.xml │ │ │ │ ├── counters.xml │ │ │ │ ├── hingecabinet.xml │ │ │ │ ├── kettle.xml │ │ │ │ ├── kitchen.xml │ │ │ │ ├── meshes │ │ │ │ │ ├── burnerplate.stl │ │ │ │ │ ├── burnerplate_mesh.stl │ │ │ │ │ ├── cabinetbase.stl │ │ │ │ │ ├── cabinetdrawer.stl │ │ │ │ │ ├── cabinethandle.stl │ │ │ │ │ ├── countertop.stl │ │ │ │ │ ├── faucet.stl │ │ │ │ │ ├── handle2.stl │ │ │ │ │ ├── hingecabinet.stl │ │ │ │ │ ├── hingedoor.stl │ │ │ │ │ ├── hingehandle.stl │ │ │ │ │ ├── hood.stl │ │ │ │ │ ├── kettle.stl │ │ │ │ │ ├── kettlehandle.stl │ │ │ │ │ ├── knob.stl │ │ │ │ │ ├── lightswitch.stl │ │ │ │ │ ├── lightswitchbase.stl │ │ │ │ │ ├── micro.stl │ │ │ │ │ ├── microbutton.stl │ │ │ │ │ ├── microdoor.stl │ │ │ │ │ ├── microefeet.stl │ │ │ │ │ ├── microfeet.stl │ │ │ │ │ ├── microhandle.stl │ │ │ │ │ ├── microwindow.stl │ │ │ │ │ ├── oven.stl │ │ │ │ │ ├── ovenhandle.stl │ │ │ │ │ ├── oventop.stl │ │ │ │ │ ├── ovenwindow.stl │ │ │ │ │ ├── slidecabinet.stl │ │ │ │ │ ├── slidedoor.stl │ │ │ │ │ ├── stoverim.stl │ │ │ │ │ ├── tile.stl │ │ │ │ │ └── wall.stl │ │ │ │ ├── microwave.xml │ │ │ │ ├── oven.xml │ │ │ │ └── slidecabinet.xml │ │ │ └── scenes │ │ │ │ └── basic_scene.xml │ │ ├── kitchen_envs.py │ │ └── third_party │ │ │ └── franka │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── assets │ │ │ ├── actuator0.xml │ │ │ ├── actuator1.xml │ │ │ ├── assets.xml │ │ │ ├── basic_scene.xml │ │ │ ├── chain0.xml │ │ │ ├── chain0_overlay.xml │ │ │ ├── chain1.xml │ │ │ └── teleop_actuator.xml │ │ │ ├── bi-franka_panda.xml │ │ │ ├── franka_panda.xml │ │ │ ├── franka_panda_teleop.xml │ │ │ └── meshes │ │ │ ├── collision │ │ │ ├── finger.stl │ │ │ ├── hand.stl │ │ │ ├── link0.stl │ │ │ ├── link1.stl │ │ │ ├── link2.stl │ │ │ ├── link3.stl │ │ │ ├── link4.stl │ │ │ ├── link5.stl │ │ │ ├── link6.stl │ │ │ └── link7.stl │ │ │ └── visual │ │ │ ├── finger.stl │ │ │ ├── hand.stl │ │ │ ├── link0.stl │ │ │ ├── link1.stl │ │ │ ├── link2.stl │ │ │ ├── link3.stl │ │ │ ├── link4.stl │ │ │ ├── link5.stl │ │ │ ├── link6.stl │ │ │ └── link7.stl │ ├── locomotion │ │ ├── __init__.py │ │ ├── ant.py │ │ ├── assets │ │ │ ├── ant.xml │ │ │ └── point.xml │ │ ├── common.py │ │ ├── generate_dataset.py │ │ ├── goal_reaching_env.py │ │ ├── maze_env.py │ │ ├── mujoco_goal_env.py │ │ ├── point.py │ │ ├── swimmer.py │ │ └── wrappers.py │ ├── offline_env.py │ ├── ope.py │ ├── pointmaze │ │ ├── __init__.py │ │ ├── dynamic_mjc.py │ │ ├── gridcraft │ │ │ ├── __init__.py │ │ │ ├── grid_env.py │ │ │ ├── grid_spec.py │ │ │ ├── utils.py │ │ │ └── wrappers.py │ │ ├── maze_model.py │ │ ├── q_iteration.py │ │ ├── shortest_paths.py │ │ └── waypoint_controller.py │ ├── pointmaze_bullet │ │ ├── __init__.py │ │ ├── bullet_maze.py │ │ └── bullet_robot.py │ └── utils │ │ ├── __init__.py │ │ ├── dataset_utils.py │ │ ├── quatmath.py │ │ ├── visualize_env.py │ │ └── wrappers.py ├── scripts │ ├── check_antmaze_datasets.py │ ├── check_bullet.py │ ├── check_envs.py │ ├── check_mujoco_datasets.py │ ├── generation │ │ ├── flow_idm.py │ │ ├── generate_ant_maze_datasets.py │ │ ├── generate_kitchen_datasets.py │ │ ├── generate_maze2d_bullet_datasets.py │ │ ├── generate_maze2d_datasets.py │ │ ├── generate_minigrid_fourroom_data.py │ │ ├── hand_dapg_combined.py │ │ ├── hand_dapg_demos.py │ │ ├── hand_dapg_jax.py │ │ ├── hand_dapg_policies.py │ │ ├── hand_dapg_random.py │ │ ├── mujoco │ │ │ ├── collect_data.py │ │ │ ├── convert_buffer.py │ │ │ ├── fix_qpos_qvel.py │ │ │ └── stitch_dataset.py │ │ ├── relabel_antmaze_rewards.py │ │ └── relabel_maze2d_rewards.py │ ├── ope_rollout.py │ ├── reference_scores │ │ ├── adroit_expert.py │ │ ├── carla_lane_controller.py │ │ ├── generate_ref_min_score.py │ │ ├── generate_ref_min_score.sh │ │ ├── maze2d_bullet_controller.py │ │ ├── maze2d_controller.py │ │ └── minigrid_controller.py │ └── visualize_dataset.py └── setup.py ├── README.md ├── locomotion ├── .dockerignore ├── .gitattributes ├── .gitignore ├── Dockerfile ├── LICENSE ├── README.md ├── Singularity.def ├── azure │ ├── config.py │ ├── download.sh │ ├── files │ │ ├── 10_nvidia.json │ │ ├── Xdummy │ │ └── mjkey.txt │ ├── launch.py │ └── sync.sh ├── config │ ├── discriminator.py │ ├── gradient_matching.py │ └── locomotion.py ├── diffuser │ ├── __init__.py │ ├── datasets │ │ ├── __init__.py │ │ ├── buffer.py │ │ ├── d4rl.py │ │ ├── normalization.py │ │ ├── preprocessing.py │ │ └── sequence.py │ ├── environments │ │ ├── __init__.py │ │ ├── ant.py │ │ ├── assets │ │ │ ├── ant.xml │ │ │ ├── half_cheetah.xml │ │ │ ├── hopper.xml │ │ │ └── walker2d.xml │ │ ├── half_cheetah.py │ │ ├── hopper.py │ │ ├── registration.py │ │ └── walker2d.py │ ├── models │ │ ├── __init__.py │ │ ├── diffusion.py │ │ ├── helpers.py │ │ └── temporal.py │ ├── rrf_diffusion │ │ ├── __init__.py │ │ ├── discriminator.py │ │ ├── discriminator_trainer.py │ │ ├── gradient_matching.py │ │ ├── gradient_matching_trainer.py │ │ ├── preprocess_dataset.py │ │ ├── train_discriminator.py │ │ └── train_gradient_matching.py │ ├── sampling │ │ ├── __init__.py │ │ ├── functions.py │ │ ├── guides.py │ │ └── policies.py │ └── utils │ │ ├── __init__.py │ │ ├── arrays.py │ │ ├── cloud.py │ │ ├── colab.py │ │ ├── config.py │ │ ├── datasets.py │ │ ├── debugging.py │ │ ├── git_utils.py │ │ ├── iql.py │ │ ├── logger.py │ │ ├── progress.py │ │ ├── pybullet_utils.py │ │ ├── rendering.py │ │ ├── serialization.py │ │ ├── setup.py │ │ ├── timer.py │ │ ├── training.py │ │ ├── transformations.py │ │ └── video.py ├── environment.yml ├── experiment_commands │ └── main_results.json ├── original_diffuser_readme.md ├── plotting │ ├── locomotion_scores.py │ └── table.py ├── quetzal.yaml ├── requirements.txt ├── run.py ├── scripts │ ├── plan_discriminator_parallel.py │ ├── plan_gradient_matching.py │ ├── plan_gradient_matching_parallel.py │ ├── plan_guided.py │ ├── train.py │ └── train_values.py ├── setup.py ├── sweep_configs │ ├── ablations │ │ └── different_seed_base.yaml │ ├── final │ │ └── final_rollouts.yaml │ └── large_sweep_v1 │ │ ├── gradient_matching │ │ ├── gradient_matching_large_sweep_L2reg_no_noise_v1.yaml │ │ ├── gradient_matching_large_sweep_L2reg_noised_v1.yaml │ │ ├── gradient_matching_large_sweep_grad_clip_no_noise_v1.yaml │ │ ├── gradient_matching_large_sweep_grad_clip_noised_v1.yaml │ │ ├── gradient_matching_large_sweep_lr_no_noise_v1.yaml │ │ ├── gradient_matching_large_sweep_lr_noised_v1.yaml │ │ ├── gradient_matching_large_sweep_weight_clip_no_noise_v1.yaml │ │ └── gradient_matching_large_sweep_weight_clip_noised_v1.yaml │ │ └── rollouts │ │ ├── rollouts_large_sweep_L2reg_no_noise_v1.yaml │ │ ├── rollouts_large_sweep_L2reg_noised_v1.yaml │ │ ├── rollouts_large_sweep_grad_clip_no_noise_v1.yaml │ │ ├── rollouts_large_sweep_grad_clip_noised_v1.yaml │ │ ├── rollouts_large_sweep_lr_no_noise_v1.yaml │ │ ├── rollouts_large_sweep_lr_noised_v1.yaml │ │ ├── rollouts_large_sweep_weight_clip_no_noise_v1.yaml │ │ └── rollouts_large_sweep_weight_clip_noised_v1.yaml └── tmp_diffusion_commands.txt ├── maze2d ├── .dockerignore ├── .gitignore ├── LICENSE ├── README.md ├── config │ ├── discriminator.py │ ├── gradient_matching.py │ ├── locomotion.py │ └── maze2d.py ├── dataset_generation │ ├── generate_commands_d4rl_datasets.py │ ├── make_json.ipynb │ ├── maze2d-large-v1.json │ ├── maze2d-medium-v1.json │ ├── maze2d-open-v0.json │ ├── maze2d-umaze-v0.json │ └── preprocessing_commands.txt ├── diffuser │ ├── __init__.py │ ├── datasets │ │ ├── __init__.py │ │ ├── buffer.py │ │ ├── d4rl.py │ │ ├── normalization.py │ │ ├── preprocessing.py │ │ └── sequence.py │ ├── environments │ │ ├── __init__.py │ │ ├── ant.py │ │ ├── assets │ │ │ ├── ant.xml │ │ │ ├── half_cheetah.xml │ │ │ ├── hopper.xml │ │ │ └── walker2d.xml │ │ ├── half_cheetah.py │ │ ├── hopper.py │ │ ├── registration.py │ │ └── walker2d.py │ ├── models │ │ ├── __init__.py │ │ ├── diffusion.py │ │ ├── helpers.py │ │ └── temporal.py │ ├── rrf_diffusion │ │ ├── __init__.py │ │ ├── depict_reward.py │ │ ├── discriminator.py │ │ ├── discriminator_trainer.py │ │ ├── gradient_matching.py │ │ ├── gradient_matching_trainer.py │ │ ├── preprocess_dataset.py │ │ ├── train_discriminator.py │ │ └── train_gradient_matching.py │ ├── sampling │ │ ├── functions.py │ │ ├── guides.py │ │ └── policies.py │ └── utils │ │ ├── __init__.py │ │ ├── arrays.py │ │ ├── cloud.py │ │ ├── colab.py │ │ ├── config.py │ │ ├── datasets.py │ │ ├── debugging.py │ │ ├── git_utils.py │ │ ├── iql.py │ │ ├── progress.py │ │ ├── pybullet_utils.py │ │ ├── rendering.py │ │ ├── serialization.py │ │ ├── setup.py │ │ ├── timer.py │ │ ├── training.py │ │ ├── transformations.py │ │ └── video.py ├── diffusion_training_commands │ ├── diffusion_commands.txt │ ├── make_json.ipynb │ ├── maze2d-large-v1.json │ ├── maze2d-medium-v1.json │ ├── maze2d-open-v0.json │ └── maze2d-umaze-v0.json ├── download_extract_dataset.sh ├── environment.yml ├── experiment_commands │ ├── maze2d-large-v1.json │ ├── maze2d-medium-v1.json │ ├── maze2d-open-v0.json │ └── maze2d-umaze-v0.json ├── installation.txt ├── original_diffuser_readme.md ├── requirements.txt ├── run.py ├── scripts │ └── train.py ├── setup.py └── sweep_configs │ ├── all_maze_experiments_10m_v1.yaml │ └── dataset_size_ablation_1m_10k_v1.yaml ├── pics └── architecture_figure.jpg └── safe-stable-diffusion ├── .gitignore ├── README.md ├── diffusers ├── .github │ ├── ISSUE_TEMPLATE │ │ ├── bug-report.yml │ │ ├── config.yml │ │ ├── feature_request.md │ │ ├── feedback.md │ │ └── new-model-addition.yml │ ├── actions │ │ └── setup-miniconda │ │ │ └── action.yml │ └── workflows │ │ ├── build_docker_images.yml │ │ ├── build_documentation.yml │ │ ├── build_pr_documentation.yml │ │ ├── delete_doc_comment.yml │ │ ├── nightly_tests.yml │ │ ├── pr_quality.yml │ │ ├── pr_tests.yml │ │ ├── push_tests.yml │ │ ├── push_tests_fast.yml │ │ ├── stale.yml │ │ └── typos.yml ├── .gitignore ├── CITATION.cff ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── Makefile ├── PHILOSOPHY.md ├── README.md ├── _typos.toml ├── docker │ ├── diffusers-flax-cpu │ │ └── Dockerfile │ ├── diffusers-flax-tpu │ │ └── Dockerfile │ ├── diffusers-onnxruntime-cpu │ │ └── Dockerfile │ ├── diffusers-onnxruntime-cuda │ │ └── Dockerfile │ ├── diffusers-pytorch-cpu │ │ └── Dockerfile │ └── diffusers-pytorch-cuda │ │ └── Dockerfile ├── docs │ ├── README.md │ ├── TRANSLATING.md │ └── source │ │ ├── _config.py │ │ ├── en │ │ ├── _toctree.yml │ │ ├── api │ │ │ ├── configuration.mdx │ │ │ ├── diffusion_pipeline.mdx │ │ │ ├── experimental │ │ │ │ └── rl.mdx │ │ │ ├── loaders.mdx │ │ │ ├── logging.mdx │ │ │ ├── models.mdx │ │ │ ├── outputs.mdx │ │ │ ├── pipelines │ │ │ │ ├── alt_diffusion.mdx │ │ │ │ ├── audio_diffusion.mdx │ │ │ │ ├── audioldm.mdx │ │ │ │ ├── cycle_diffusion.mdx │ │ │ │ ├── dance_diffusion.mdx │ │ │ │ ├── ddim.mdx │ │ │ │ ├── ddpm.mdx │ │ │ │ ├── dit.mdx │ │ │ │ ├── if.mdx │ │ │ │ ├── latent_diffusion.mdx │ │ │ │ ├── latent_diffusion_uncond.mdx │ │ │ │ ├── overview.mdx │ │ │ │ ├── paint_by_example.mdx │ │ │ │ ├── pndm.mdx │ │ │ │ ├── repaint.mdx │ │ │ │ ├── score_sde_ve.mdx │ │ │ │ ├── semantic_stable_diffusion.mdx │ │ │ │ ├── spectrogram_diffusion.mdx │ │ │ │ ├── stable_diffusion │ │ │ │ │ ├── attend_and_excite.mdx │ │ │ │ │ ├── controlnet.mdx │ │ │ │ │ ├── depth2img.mdx │ │ │ │ │ ├── image_variation.mdx │ │ │ │ │ ├── img2img.mdx │ │ │ │ │ ├── inpaint.mdx │ │ │ │ │ ├── latent_upscale.mdx │ │ │ │ │ ├── model_editing.mdx │ │ │ │ │ ├── overview.mdx │ │ │ │ │ ├── panorama.mdx │ │ │ │ │ ├── pix2pix.mdx │ │ │ │ │ ├── pix2pix_zero.mdx │ │ │ │ │ ├── self_attention_guidance.mdx │ │ │ │ │ ├── text2img.mdx │ │ │ │ │ └── upscale.mdx │ │ │ │ ├── stable_diffusion_2.mdx │ │ │ │ ├── stable_diffusion_safe.mdx │ │ │ │ ├── stable_unclip.mdx │ │ │ │ ├── stochastic_karras_ve.mdx │ │ │ │ ├── text_to_video.mdx │ │ │ │ ├── text_to_video_zero.mdx │ │ │ │ ├── unclip.mdx │ │ │ │ ├── versatile_diffusion.mdx │ │ │ │ └── vq_diffusion.mdx │ │ │ └── schedulers │ │ │ │ ├── ddim.mdx │ │ │ │ ├── ddim_inverse.mdx │ │ │ │ ├── ddpm.mdx │ │ │ │ ├── deis.mdx │ │ │ │ ├── dpm_discrete.mdx │ │ │ │ ├── dpm_discrete_ancestral.mdx │ │ │ │ ├── euler.mdx │ │ │ │ ├── euler_ancestral.mdx │ │ │ │ ├── heun.mdx │ │ │ │ ├── ipndm.mdx │ │ │ │ ├── lms_discrete.mdx │ │ │ │ ├── multistep_dpm_solver.mdx │ │ │ │ ├── overview.mdx │ │ │ │ ├── pndm.mdx │ │ │ │ ├── repaint.mdx │ │ │ │ ├── score_sde_ve.mdx │ │ │ │ ├── score_sde_vp.mdx │ │ │ │ ├── singlestep_dpm_solver.mdx │ │ │ │ ├── stochastic_karras_ve.mdx │ │ │ │ ├── unipc.mdx │ │ │ │ └── vq_diffusion.mdx │ │ ├── conceptual │ │ │ ├── contribution.mdx │ │ │ ├── ethical_guidelines.mdx │ │ │ ├── evaluation.mdx │ │ │ └── philosophy.mdx │ │ ├── imgs │ │ │ ├── access_request.png │ │ │ └── diffusers_library.jpg │ │ ├── index.mdx │ │ ├── installation.mdx │ │ ├── optimization │ │ │ ├── coreml.mdx │ │ │ ├── fp16.mdx │ │ │ ├── habana.mdx │ │ │ ├── mps.mdx │ │ │ ├── onnx.mdx │ │ │ ├── open_vino.mdx │ │ │ ├── opt_overview.mdx │ │ │ ├── tome.mdx │ │ │ ├── torch2.0.mdx │ │ │ └── xformers.mdx │ │ ├── quicktour.mdx │ │ ├── stable_diffusion.mdx │ │ ├── training │ │ │ ├── controlnet.mdx │ │ │ ├── custom_diffusion.mdx │ │ │ ├── dreambooth.mdx │ │ │ ├── instructpix2pix.mdx │ │ │ ├── lora.mdx │ │ │ ├── overview.mdx │ │ │ ├── text2image.mdx │ │ │ ├── text_inversion.mdx │ │ │ └── unconditional_training.mdx │ │ ├── tutorials │ │ │ ├── basic_training.mdx │ │ │ └── tutorial_overview.mdx │ │ └── using-diffusers │ │ │ ├── audio.mdx │ │ │ ├── conditional_image_generation.mdx │ │ │ ├── contribute_pipeline.mdx │ │ │ ├── controlling_generation.mdx │ │ │ ├── custom_pipeline_examples.mdx │ │ │ ├── custom_pipeline_overview.mdx │ │ │ ├── depth2img.mdx │ │ │ ├── img2img.mdx │ │ │ ├── inpaint.mdx │ │ │ ├── kerascv.mdx │ │ │ ├── loading.mdx │ │ │ ├── loading_overview.mdx │ │ │ ├── other-modalities.mdx │ │ │ ├── pipeline_overview.mdx │ │ │ ├── reproducibility.mdx │ │ │ ├── reusing_seeds.mdx │ │ │ ├── rl.mdx │ │ │ ├── schedulers.mdx │ │ │ ├── stable_diffusion_jax_how_to.mdx │ │ │ ├── unconditional_image_generation.mdx │ │ │ ├── using_safetensors │ │ │ ├── using_safetensors.mdx │ │ │ ├── weighted_prompts.mdx │ │ │ └── write_own_pipeline.mdx │ │ ├── ko │ │ ├── _toctree.yml │ │ ├── in_translation.mdx │ │ ├── index.mdx │ │ ├── installation.mdx │ │ └── quicktour.mdx │ │ └── zh │ │ ├── _toctree.yml │ │ ├── index.mdx │ │ ├── installation.mdx │ │ └── quicktour.mdx ├── examples │ ├── README.md │ ├── community │ │ ├── README.md │ │ ├── bit_diffusion.py │ │ ├── checkpoint_merger.py │ │ ├── clip_guided_stable_diffusion.py │ │ ├── clip_guided_stable_diffusion_img2img.py │ │ ├── composable_stable_diffusion.py │ │ ├── ddim_noise_comparative_analysis.py │ │ ├── imagic_stable_diffusion.py │ │ ├── img2img_inpainting.py │ │ ├── interpolate_stable_diffusion.py │ │ ├── lpw_stable_diffusion.py │ │ ├── lpw_stable_diffusion_onnx.py │ │ ├── magic_mix.py │ │ ├── multilingual_stable_diffusion.py │ │ ├── one_step_unet.py │ │ ├── sd_text2img_k_diffusion.py │ │ ├── seed_resize_stable_diffusion.py │ │ ├── speech_to_image_diffusion.py │ │ ├── stable_diffusion_comparison.py │ │ ├── stable_diffusion_controlnet_img2img.py │ │ ├── stable_diffusion_controlnet_inpaint.py │ │ ├── stable_diffusion_controlnet_inpaint_img2img.py │ │ ├── stable_diffusion_mega.py │ │ ├── stable_diffusion_tensorrt_txt2img.py │ │ ├── stable_unclip.py │ │ ├── text_inpainting.py │ │ ├── tiled_upscaling.py │ │ ├── unclip_image_interpolation.py │ │ ├── unclip_text_interpolation.py │ │ └── wildcard_stable_diffusion.py │ ├── conftest.py │ ├── controlnet │ │ ├── README.md │ │ ├── requirements.txt │ │ ├── requirements_flax.txt │ │ ├── train_controlnet.py │ │ └── train_controlnet_flax.py │ ├── custom_diffusion │ │ ├── README.md │ │ ├── requirements.txt │ │ ├── retrieve.py │ │ └── train_custom_diffusion.py │ ├── dreambooth │ │ ├── README.md │ │ ├── requirements.txt │ │ ├── requirements_flax.txt │ │ ├── train_dreambooth.py │ │ ├── train_dreambooth_flax.py │ │ └── train_dreambooth_lora.py │ ├── inference │ │ ├── README.md │ │ ├── image_to_image.py │ │ └── inpainting.py │ ├── instruct_pix2pix │ │ ├── README.md │ │ ├── requirements.txt │ │ └── train_instruct_pix2pix.py │ ├── research_projects │ │ ├── README.md │ │ ├── colossalai │ │ │ ├── README.md │ │ │ ├── inference.py │ │ │ ├── requirement.txt │ │ │ └── train_dreambooth_colossalai.py │ │ ├── dreambooth_inpaint │ │ │ ├── README.md │ │ │ ├── requirements.txt │ │ │ ├── train_dreambooth_inpaint.py │ │ │ └── train_dreambooth_inpaint_lora.py │ │ ├── intel_opts │ │ │ ├── README.md │ │ │ ├── inference_bf16.py │ │ │ ├── textual_inversion │ │ │ │ ├── README.md │ │ │ │ ├── requirements.txt │ │ │ │ └── textual_inversion_bf16.py │ │ │ └── textual_inversion_dfq │ │ │ │ ├── README.md │ │ │ │ ├── requirements.txt │ │ │ │ ├── text2images.py │ │ │ │ └── textual_inversion.py │ │ ├── lora │ │ │ ├── README.md │ │ │ ├── requirements.txt │ │ │ └── train_text_to_image_lora.py │ │ ├── mulit_token_textual_inversion │ │ │ ├── README.md │ │ │ ├── multi_token_clip.py │ │ │ ├── requirements.txt │ │ │ ├── requirements_flax.txt │ │ │ ├── textual_inversion.py │ │ │ └── textual_inversion_flax.py │ │ ├── multi_subject_dreambooth │ │ │ ├── README.md │ │ │ ├── requirements.txt │ │ │ └── train_multi_subject_dreambooth.py │ │ └── onnxruntime │ │ │ ├── README.md │ │ │ ├── text_to_image │ │ │ ├── README.md │ │ │ ├── requirements.txt │ │ │ └── train_text_to_image.py │ │ │ ├── textual_inversion │ │ │ ├── README.md │ │ │ ├── requirements.txt │ │ │ └── textual_inversion.py │ │ │ └── unconditional_image_generation │ │ │ ├── README.md │ │ │ ├── requirements.txt │ │ │ └── train_unconditional.py │ ├── rl │ │ ├── README.md │ │ └── run_diffuser_locomotion.py │ ├── test_examples.py │ ├── text_to_image │ │ ├── README.md │ │ ├── requirements.txt │ │ ├── requirements_flax.txt │ │ ├── train_text_to_image.py │ │ ├── train_text_to_image_flax.py │ │ └── train_text_to_image_lora.py │ ├── textual_inversion │ │ ├── README.md │ │ ├── requirements.txt │ │ ├── requirements_flax.txt │ │ ├── textual_inversion.py │ │ └── textual_inversion_flax.py │ └── unconditional_image_generation │ │ ├── README.md │ │ ├── requirements.txt │ │ └── train_unconditional.py ├── pyproject.toml ├── scripts │ ├── __init__.py │ ├── change_naming_configs_and_checkpoints.py │ ├── conversion_ldm_uncond.py │ ├── convert_dance_diffusion_to_diffusers.py │ ├── convert_ddpm_original_checkpoint_to_diffusers.py │ ├── convert_diffusers_to_original_stable_diffusion.py │ ├── convert_dit_to_diffusers.py │ ├── convert_if.py │ ├── convert_k_upscaler_to_diffusers.py │ ├── convert_kakao_brain_unclip_to_diffusers.py │ ├── convert_ldm_original_checkpoint_to_diffusers.py │ ├── convert_lora_safetensor_to_diffusers.py │ ├── convert_models_diffuser_to_diffusers.py │ ├── convert_ms_text_to_video_to_diffusers.py │ ├── convert_music_spectrogram_to_diffusers.py │ ├── convert_ncsnpp_original_checkpoint_to_diffusers.py │ ├── convert_original_audioldm_to_diffusers.py │ ├── convert_original_controlnet_to_diffusers.py │ ├── convert_original_stable_diffusion_to_diffusers.py │ ├── convert_stable_diffusion_checkpoint_to_onnx.py │ ├── convert_unclip_txt2img_to_image_variation.py │ ├── convert_vae_diff_to_onnx.py │ ├── convert_vae_pt_to_diffusers.py │ ├── convert_versatile_diffusion_to_diffusers.py │ ├── convert_vq_diffusion_to_diffusers.py │ └── generate_logits.py ├── setup.cfg ├── setup.py ├── src │ └── diffusers │ │ ├── __init__.py │ │ ├── commands │ │ ├── __init__.py │ │ ├── diffusers_cli.py │ │ └── env.py │ │ ├── configuration_utils.py │ │ ├── dependency_versions_check.py │ │ ├── dependency_versions_table.py │ │ ├── experimental │ │ ├── README.md │ │ ├── __init__.py │ │ └── rl │ │ │ ├── __init__.py │ │ │ └── value_guided_sampling.py │ │ ├── image_processor.py │ │ ├── loaders.py │ │ ├── models │ │ ├── README.md │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── attention_flax.py │ │ ├── attention_processor.py │ │ ├── autoencoder_kl.py │ │ ├── controlnet.py │ │ ├── controlnet_flax.py │ │ ├── cross_attention.py │ │ ├── dual_transformer_2d.py │ │ ├── embeddings.py │ │ ├── embeddings_flax.py │ │ ├── modeling_flax_pytorch_utils.py │ │ ├── modeling_flax_utils.py │ │ ├── modeling_pytorch_flax_utils.py │ │ ├── modeling_utils.py │ │ ├── prior_transformer.py │ │ ├── resnet.py │ │ ├── resnet_flax.py │ │ ├── t5_film_transformer.py │ │ ├── transformer_2d.py │ │ ├── transformer_temporal.py │ │ ├── unet_1d.py │ │ ├── unet_1d_blocks.py │ │ ├── unet_2d.py │ │ ├── unet_2d_blocks.py │ │ ├── unet_2d_blocks_flax.py │ │ ├── unet_2d_condition.py │ │ ├── unet_2d_condition_flax.py │ │ ├── unet_3d_blocks.py │ │ ├── unet_3d_condition.py │ │ ├── vae.py │ │ ├── vae_flax.py │ │ └── vq_model.py │ │ ├── optimization.py │ │ ├── pipeline_utils.py │ │ ├── pipelines │ │ ├── README.md │ │ ├── __init__.py │ │ ├── alt_diffusion │ │ │ ├── __init__.py │ │ │ ├── modeling_roberta_series.py │ │ │ ├── pipeline_alt_diffusion.py │ │ │ └── pipeline_alt_diffusion_img2img.py │ │ ├── audio_diffusion │ │ │ ├── __init__.py │ │ │ ├── mel.py │ │ │ └── pipeline_audio_diffusion.py │ │ ├── audioldm │ │ │ ├── __init__.py │ │ │ └── pipeline_audioldm.py │ │ ├── dance_diffusion │ │ │ ├── __init__.py │ │ │ └── pipeline_dance_diffusion.py │ │ ├── ddim │ │ │ ├── __init__.py │ │ │ └── pipeline_ddim.py │ │ ├── ddpm │ │ │ ├── __init__.py │ │ │ └── pipeline_ddpm.py │ │ ├── deepfloyd_if │ │ │ ├── __init__.py │ │ │ ├── pipeline_if.py │ │ │ ├── pipeline_if_img2img.py │ │ │ ├── pipeline_if_img2img_superresolution.py │ │ │ ├── pipeline_if_inpainting.py │ │ │ ├── pipeline_if_inpainting_superresolution.py │ │ │ ├── pipeline_if_superresolution.py │ │ │ ├── safety_checker.py │ │ │ ├── timesteps.py │ │ │ └── watermark.py │ │ ├── dit │ │ │ ├── __init__.py │ │ │ └── pipeline_dit.py │ │ ├── latent_diffusion │ │ │ ├── __init__.py │ │ │ ├── pipeline_latent_diffusion.py │ │ │ └── pipeline_latent_diffusion_superresolution.py │ │ ├── latent_diffusion_uncond │ │ │ ├── __init__.py │ │ │ └── pipeline_latent_diffusion_uncond.py │ │ ├── onnx_utils.py │ │ ├── paint_by_example │ │ │ ├── __init__.py │ │ │ ├── image_encoder.py │ │ │ └── pipeline_paint_by_example.py │ │ ├── pipeline_flax_utils.py │ │ ├── pipeline_utils.py │ │ ├── pndm │ │ │ ├── __init__.py │ │ │ └── pipeline_pndm.py │ │ ├── repaint │ │ │ ├── __init__.py │ │ │ └── pipeline_repaint.py │ │ ├── score_sde_ve │ │ │ ├── __init__.py │ │ │ └── pipeline_score_sde_ve.py │ │ ├── semantic_stable_diffusion │ │ │ ├── __init__.py │ │ │ └── pipeline_semantic_stable_diffusion.py │ │ ├── spectrogram_diffusion │ │ │ ├── __init__.py │ │ │ ├── continous_encoder.py │ │ │ ├── midi_utils.py │ │ │ ├── notes_encoder.py │ │ │ └── pipeline_spectrogram_diffusion.py │ │ ├── stable_diffusion │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── convert_from_ckpt.py │ │ │ ├── pipeline_cycle_diffusion.py │ │ │ ├── pipeline_flax_stable_diffusion.py │ │ │ ├── pipeline_flax_stable_diffusion_controlnet.py │ │ │ ├── pipeline_flax_stable_diffusion_img2img.py │ │ │ ├── pipeline_flax_stable_diffusion_inpaint.py │ │ │ ├── pipeline_onnx_stable_diffusion.py │ │ │ ├── pipeline_onnx_stable_diffusion_img2img.py │ │ │ ├── pipeline_onnx_stable_diffusion_inpaint.py │ │ │ ├── pipeline_onnx_stable_diffusion_inpaint_legacy.py │ │ │ ├── pipeline_onnx_stable_diffusion_upscale.py │ │ │ ├── pipeline_stable_diffusion.py │ │ │ ├── pipeline_stable_diffusion_attend_and_excite.py │ │ │ ├── pipeline_stable_diffusion_controlnet.py │ │ │ ├── pipeline_stable_diffusion_depth2img.py │ │ │ ├── pipeline_stable_diffusion_image_variation.py │ │ │ ├── pipeline_stable_diffusion_img2img.py │ │ │ ├── pipeline_stable_diffusion_inpaint.py │ │ │ ├── pipeline_stable_diffusion_inpaint_legacy.py │ │ │ ├── pipeline_stable_diffusion_instruct_pix2pix.py │ │ │ ├── pipeline_stable_diffusion_k_diffusion.py │ │ │ ├── pipeline_stable_diffusion_latent_upscale.py │ │ │ ├── pipeline_stable_diffusion_model_editing.py │ │ │ ├── pipeline_stable_diffusion_panorama.py │ │ │ ├── pipeline_stable_diffusion_pix2pix_zero.py │ │ │ ├── pipeline_stable_diffusion_sag.py │ │ │ ├── pipeline_stable_diffusion_upscale.py │ │ │ ├── pipeline_stable_unclip.py │ │ │ ├── pipeline_stable_unclip_img2img.py │ │ │ ├── safety_checker.py │ │ │ ├── safety_checker_flax.py │ │ │ └── stable_unclip_image_normalizer.py │ │ ├── stable_diffusion_safe │ │ │ ├── __init__.py │ │ │ ├── pipeline_stable_diffusion_safe.py │ │ │ └── safety_checker.py │ │ ├── stochastic_karras_ve │ │ │ ├── __init__.py │ │ │ └── pipeline_stochastic_karras_ve.py │ │ ├── text_to_video_synthesis │ │ │ ├── __init__.py │ │ │ ├── pipeline_text_to_video_synth.py │ │ │ └── pipeline_text_to_video_zero.py │ │ ├── unclip │ │ │ ├── __init__.py │ │ │ ├── pipeline_unclip.py │ │ │ ├── pipeline_unclip_image_variation.py │ │ │ └── text_proj.py │ │ ├── versatile_diffusion │ │ │ ├── __init__.py │ │ │ ├── modeling_text_unet.py │ │ │ ├── pipeline_versatile_diffusion.py │ │ │ ├── pipeline_versatile_diffusion_dual_guided.py │ │ │ ├── pipeline_versatile_diffusion_image_variation.py │ │ │ └── pipeline_versatile_diffusion_text_to_image.py │ │ └── vq_diffusion │ │ │ ├── __init__.py │ │ │ └── pipeline_vq_diffusion.py │ │ ├── schedulers │ │ ├── README.md │ │ ├── __init__.py │ │ ├── scheduling_ddim.py │ │ ├── scheduling_ddim_flax.py │ │ ├── scheduling_ddim_inverse.py │ │ ├── scheduling_ddpm.py │ │ ├── scheduling_ddpm_flax.py │ │ ├── scheduling_deis_multistep.py │ │ ├── scheduling_dpmsolver_multistep.py │ │ ├── scheduling_dpmsolver_multistep_flax.py │ │ ├── scheduling_dpmsolver_singlestep.py │ │ ├── scheduling_euler_ancestral_discrete.py │ │ ├── scheduling_euler_discrete.py │ │ ├── scheduling_heun_discrete.py │ │ ├── scheduling_ipndm.py │ │ ├── scheduling_k_dpm_2_ancestral_discrete.py │ │ ├── scheduling_k_dpm_2_discrete.py │ │ ├── scheduling_karras_ve.py │ │ ├── scheduling_karras_ve_flax.py │ │ ├── scheduling_lms_discrete.py │ │ ├── scheduling_lms_discrete_flax.py │ │ ├── scheduling_pndm.py │ │ ├── scheduling_pndm_flax.py │ │ ├── scheduling_repaint.py │ │ ├── scheduling_sde_ve.py │ │ ├── scheduling_sde_ve_flax.py │ │ ├── scheduling_sde_vp.py │ │ ├── scheduling_unclip.py │ │ ├── scheduling_unipc_multistep.py │ │ ├── scheduling_utils.py │ │ ├── scheduling_utils_flax.py │ │ └── scheduling_vq_diffusion.py │ │ ├── training_utils.py │ │ └── utils │ │ ├── __init__.py │ │ ├── accelerate_utils.py │ │ ├── constants.py │ │ ├── deprecation_utils.py │ │ ├── doc_utils.py │ │ ├── dummy_flax_and_transformers_objects.py │ │ ├── dummy_flax_objects.py │ │ ├── dummy_note_seq_objects.py │ │ ├── dummy_onnx_objects.py │ │ ├── dummy_pt_objects.py │ │ ├── dummy_torch_and_librosa_objects.py │ │ ├── dummy_torch_and_scipy_objects.py │ │ ├── dummy_torch_and_transformers_and_k_diffusion_objects.py │ │ ├── dummy_torch_and_transformers_and_onnx_objects.py │ │ ├── dummy_torch_and_transformers_objects.py │ │ ├── dummy_transformers_and_torch_and_note_seq_objects.py │ │ ├── dynamic_modules_utils.py │ │ ├── hub_utils.py │ │ ├── import_utils.py │ │ ├── logging.py │ │ ├── model_card_template.md │ │ ├── outputs.py │ │ ├── pil_utils.py │ │ ├── testing_utils.py │ │ └── torch_utils.py ├── tests │ ├── __init__.py │ ├── conftest.py │ ├── fixtures │ │ ├── custom_pipeline │ │ │ ├── pipeline.py │ │ │ └── what_ever.py │ │ └── elise_format0.mid │ ├── models │ │ ├── __init__.py │ │ ├── test_attention_processor.py │ │ ├── test_layers_utils.py │ │ ├── test_lora_layers.py │ │ ├── test_modeling_common.py │ │ ├── test_modeling_common_flax.py │ │ ├── test_models_unet_1d.py │ │ ├── test_models_unet_2d.py │ │ ├── test_models_unet_2d_condition.py │ │ ├── test_models_unet_2d_flax.py │ │ ├── test_models_unet_3d_condition.py │ │ ├── test_models_vae.py │ │ ├── test_models_vae_flax.py │ │ ├── test_models_vq.py │ │ ├── test_unet_2d_blocks.py │ │ └── test_unet_blocks_common.py │ ├── others │ │ ├── test_check_copies.py │ │ ├── test_check_dummies.py │ │ ├── test_config.py │ │ ├── test_ema.py │ │ ├── test_hub_utils.py │ │ ├── test_image_processor.py │ │ ├── test_outputs.py │ │ ├── test_training.py │ │ └── test_utils.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── altdiffusion │ │ │ ├── __init__.py │ │ │ ├── test_alt_diffusion.py │ │ │ └── test_alt_diffusion_img2img.py │ │ ├── audio_diffusion │ │ │ ├── __init__.py │ │ │ └── test_audio_diffusion.py │ │ ├── audioldm │ │ │ ├── __init__.py │ │ │ └── test_audioldm.py │ │ ├── dance_diffusion │ │ │ ├── __init__.py │ │ │ └── test_dance_diffusion.py │ │ ├── ddim │ │ │ ├── __init__.py │ │ │ └── test_ddim.py │ │ ├── ddpm │ │ │ ├── __init__.py │ │ │ └── test_ddpm.py │ │ ├── deepfloyd_if │ │ │ ├── __init__.py │ │ │ ├── test_if.py │ │ │ ├── test_if_img2img.py │ │ │ ├── test_if_img2img_superresolution.py │ │ │ ├── test_if_inpainting.py │ │ │ ├── test_if_inpainting_superresolution.py │ │ │ └── test_if_superresolution.py │ │ ├── dit │ │ │ ├── __init__.py │ │ │ └── test_dit.py │ │ ├── karras_ve │ │ │ ├── __init__.py │ │ │ └── test_karras_ve.py │ │ ├── latent_diffusion │ │ │ ├── __init__.py │ │ │ ├── test_latent_diffusion.py │ │ │ ├── test_latent_diffusion_superresolution.py │ │ │ └── test_latent_diffusion_uncond.py │ │ ├── paint_by_example │ │ │ ├── __init__.py │ │ │ └── test_paint_by_example.py │ │ ├── pipeline_params.py │ │ ├── pndm │ │ │ ├── __init__.py │ │ │ └── test_pndm.py │ │ ├── repaint │ │ │ ├── __init__.py │ │ │ └── test_repaint.py │ │ ├── score_sde_ve │ │ │ ├── __init__.py │ │ │ └── test_score_sde_ve.py │ │ ├── semantic_stable_diffusion │ │ │ ├── __init__.py │ │ │ └── test_semantic_diffusion.py │ │ ├── spectrogram_diffusion │ │ │ ├── __init__.py │ │ │ └── test_spectrogram_diffusion.py │ │ ├── stable_diffusion │ │ │ ├── __init__.py │ │ │ ├── test_cycle_diffusion.py │ │ │ ├── test_onnx_stable_diffusion.py │ │ │ ├── test_onnx_stable_diffusion_img2img.py │ │ │ ├── test_onnx_stable_diffusion_inpaint.py │ │ │ ├── test_onnx_stable_diffusion_inpaint_legacy.py │ │ │ ├── test_onnx_stable_diffusion_upscale.py │ │ │ ├── test_stable_diffusion.py │ │ │ ├── test_stable_diffusion_controlnet.py │ │ │ ├── test_stable_diffusion_flax_controlnet.py │ │ │ ├── test_stable_diffusion_image_variation.py │ │ │ ├── test_stable_diffusion_img2img.py │ │ │ ├── test_stable_diffusion_inpaint.py │ │ │ ├── test_stable_diffusion_inpaint_legacy.py │ │ │ ├── test_stable_diffusion_instruction_pix2pix.py │ │ │ ├── test_stable_diffusion_k_diffusion.py │ │ │ ├── test_stable_diffusion_model_editing.py │ │ │ ├── test_stable_diffusion_panorama.py │ │ │ ├── test_stable_diffusion_pix2pix_zero.py │ │ │ └── test_stable_diffusion_sag.py │ │ ├── stable_diffusion_2 │ │ │ ├── __init__.py │ │ │ ├── test_stable_diffusion.py │ │ │ ├── test_stable_diffusion_attend_and_excite.py │ │ │ ├── test_stable_diffusion_depth.py │ │ │ ├── test_stable_diffusion_flax.py │ │ │ ├── test_stable_diffusion_flax_inpaint.py │ │ │ ├── test_stable_diffusion_inpaint.py │ │ │ ├── test_stable_diffusion_latent_upscale.py │ │ │ ├── test_stable_diffusion_upscale.py │ │ │ └── test_stable_diffusion_v_pred.py │ │ ├── stable_diffusion_safe │ │ │ ├── __init__.py │ │ │ └── test_safe_diffusion.py │ │ ├── stable_unclip │ │ │ ├── __init__.py │ │ │ ├── test_stable_unclip.py │ │ │ └── test_stable_unclip_img2img.py │ │ ├── test_pipeline_utils.py │ │ ├── test_pipelines.py │ │ ├── test_pipelines_common.py │ │ ├── test_pipelines_flax.py │ │ ├── test_pipelines_onnx_common.py │ │ ├── text_to_video │ │ │ ├── __init__.py │ │ │ ├── test_text_to_video.py │ │ │ └── test_text_to_video_zero.py │ │ ├── unclip │ │ │ ├── __init__.py │ │ │ ├── test_unclip.py │ │ │ └── test_unclip_image_variation.py │ │ ├── versatile_diffusion │ │ │ ├── __init__.py │ │ │ ├── test_versatile_diffusion_dual_guided.py │ │ │ ├── test_versatile_diffusion_image_variation.py │ │ │ ├── test_versatile_diffusion_mega.py │ │ │ └── test_versatile_diffusion_text_to_image.py │ │ └── vq_diffusion │ │ │ ├── __init__.py │ │ │ └── test_vq_diffusion.py │ └── schedulers │ │ ├── __init__.py │ │ ├── test_scheduler_ddim.py │ │ ├── test_scheduler_ddpm.py │ │ ├── test_scheduler_deis.py │ │ ├── test_scheduler_dpm_multi.py │ │ ├── test_scheduler_dpm_single.py │ │ ├── test_scheduler_euler.py │ │ ├── test_scheduler_euler_ancestral.py │ │ ├── test_scheduler_flax.py │ │ ├── test_scheduler_heun.py │ │ ├── test_scheduler_ipndm.py │ │ ├── test_scheduler_kdpm2_ancestral.py │ │ ├── test_scheduler_kdpm2_discrete.py │ │ ├── test_scheduler_lms.py │ │ ├── test_scheduler_pndm.py │ │ ├── test_scheduler_score_sde_ve.py │ │ ├── test_scheduler_unclip.py │ │ ├── test_scheduler_unipc.py │ │ ├── test_scheduler_vq_diffusion.py │ │ └── test_schedulers.py └── utils │ ├── check_config_docstrings.py │ ├── check_copies.py │ ├── check_doc_toc.py │ ├── check_dummies.py │ ├── check_inits.py │ ├── check_repo.py │ ├── check_table.py │ ├── custom_init_isort.py │ ├── get_modified_files.py │ ├── overwrite_expected_slice.py │ ├── print_env.py │ ├── release.py │ └── stale.py ├── download_extract_safe_data.sh ├── notebooks └── rank_and_decode.ipynb ├── requirements.txt ├── rrf_diffusion ├── __init__.py ├── dataset.py ├── gradient_matching.py ├── gradient_matching_trainer.py ├── models.py ├── utils.py └── utils │ ├── git_utils.py │ ├── serialization.py │ └── setup.py ├── scripts ├── train_gradient_discriminator.py └── train_gradient_matching.py └── utils ├── generate_latents.py └── merge_latents.py /D4RL/.github/FUNDING.yml: -------------------------------------------------------------------------------- 1 | github: Farama-Foundation 2 | -------------------------------------------------------------------------------- /D4RL/.github/ISSUE_TEMPLATE/bug.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/.github/ISSUE_TEMPLATE/bug.md -------------------------------------------------------------------------------- /D4RL/.github/ISSUE_TEMPLATE/proposal.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/.github/ISSUE_TEMPLATE/proposal.md -------------------------------------------------------------------------------- /D4RL/.github/ISSUE_TEMPLATE/question.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/.github/ISSUE_TEMPLATE/question.md -------------------------------------------------------------------------------- /D4RL/.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /D4RL/.github/stale.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/.github/stale.yml -------------------------------------------------------------------------------- /D4RL/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/.gitignore -------------------------------------------------------------------------------- /D4RL/CODE_OF_CONDUCT.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/CODE_OF_CONDUCT.rst -------------------------------------------------------------------------------- /D4RL/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/LICENSE -------------------------------------------------------------------------------- /D4RL/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/MANIFEST.in -------------------------------------------------------------------------------- /D4RL/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/README.md -------------------------------------------------------------------------------- /D4RL/d4rl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/carla/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/carla/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/carla/carla_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/carla/carla_env.py -------------------------------------------------------------------------------- /D4RL/d4rl/carla/data_collection_agent_lane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/carla/data_collection_agent_lane.py -------------------------------------------------------------------------------- /D4RL/d4rl/carla/data_collection_town.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/carla/data_collection_town.py -------------------------------------------------------------------------------- /D4RL/d4rl/carla/town_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/carla/town_agent.py -------------------------------------------------------------------------------- /D4RL/d4rl/flow/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/flow/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/flow/bottleneck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/flow/bottleneck.py -------------------------------------------------------------------------------- /D4RL/d4rl/flow/merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/flow/merge.py -------------------------------------------------------------------------------- /D4RL/d4rl/flow/traffic_light_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/flow/traffic_light_grid.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_bullet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_bullet/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_bullet/gym_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_bullet/gym_envs.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_minigrid/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_minigrid/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_minigrid/envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_minigrid/envs/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_minigrid/envs/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_minigrid/envs/empty.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_minigrid/envs/fourrooms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_minigrid/envs/fourrooms.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_minigrid/fourroom_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_minigrid/fourroom_controller.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_minigrid/minigrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_minigrid/minigrid.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_minigrid/register.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_minigrid/register.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_minigrid/rendering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_minigrid/rendering.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_minigrid/roomgrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_minigrid/roomgrid.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_minigrid/window.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_minigrid/window.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_minigrid/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_minigrid/wrappers.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_mujoco/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_mujoco/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/gym_mujoco/gym_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/gym_mujoco/gym_envs.py -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/.gitignore: -------------------------------------------------------------------------------- 1 | *.DS_Store 2 | -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/Adroit_hand.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/Adroit_hand.xml -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/Adroit_hand_withOverlay.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/Adroit_hand_withOverlay.xml -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/LICENSE -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/README.md -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/gallery/news.JPG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/gallery/news.JPG -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/gallery/projects.JPG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/gallery/projects.JPG -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/assets.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/assets.xml -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/chain.xml -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/chain1.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/chain1.xml -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/F1.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/F1.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/F2.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/F2.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/F3.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/F3.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/TH1_z.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/TH1_z.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/TH2_z.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/TH2_z.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/TH3_z.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/TH3_z.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/arm_base.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/arm_base.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/arm_trunk.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/arm_trunk.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/elbow_flex.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/elbow_flex.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/knuckle.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/knuckle.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/lfmetacarpal.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/lfmetacarpal.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/palm.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/palm.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/upper_arm.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/upper_arm.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/upper_arm_ass.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/upper_arm_ass.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/wrist.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/Adroit/resources/meshes/wrist.stl -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/assets/DAPG_Adroit.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/assets/DAPG_Adroit.xml -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/assets/DAPG_assets.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/assets/DAPG_assets.xml -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/assets/DAPG_door.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/assets/DAPG_door.xml -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/assets/DAPG_hammer.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/assets/DAPG_hammer.xml -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/assets/DAPG_pen.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/assets/DAPG_pen.xml -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/assets/DAPG_relocate.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/assets/DAPG_relocate.xml -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/assets/tasks.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/assets/tasks.jpg -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/door_v0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/door_v0.py -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/hammer_v0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/hammer_v0.py -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/pen_v0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/pen_v0.py -------------------------------------------------------------------------------- /D4RL/d4rl/hand_manipulation_suite/relocate_v0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/hand_manipulation_suite/relocate_v0.py -------------------------------------------------------------------------------- /D4RL/d4rl/infos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/infos.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/.pylintrc -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/.style.yapf -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/base_robot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/base_robot.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/franka/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/franka/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/franka/assets/franka_kitchen_jntpos_act_ab.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/franka/assets/franka_kitchen_jntpos_act_ab.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/franka/kitchen_multitask_v0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/franka/kitchen_multitask_v0.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/franka/robot/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/franka/robot/franka_config.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/franka/robot/franka_config.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/franka/robot/franka_robot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/franka/robot/franka_robot.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/mujoco_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/mujoco_env.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/robot_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/robot_env.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/simulation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/simulation/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/simulation/module.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/simulation/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/simulation/renderer.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/simulation/sim_robot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/simulation/sim_robot.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/utils/config.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/utils/configurable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/utils/configurable.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/utils/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/utils/constants.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/utils/parse_demos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/utils/parse_demos.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_envs/utils/quatmath.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_envs/utils/quatmath.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/.gitignore -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/CONTRIBUTING.public.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/CONTRIBUTING.public.md -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/LICENSE -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/README.public.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/README.public.md -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/backwall_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/backwall_asset.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/backwall_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/backwall_chain.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/counters_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/counters_asset.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/counters_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/counters_chain.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/hingecabinet_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/hingecabinet_asset.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/hingecabinet_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/hingecabinet_chain.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/kettle_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/kettle_asset.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/kettle_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/kettle_chain.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/microwave_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/microwave_asset.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/microwave_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/microwave_chain.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/oven_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/oven_asset.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/oven_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/oven_chain.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/slidecabinet_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/slidecabinet_asset.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/assets/slidecabinet_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/assets/slidecabinet_chain.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/counters.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/counters.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/hingecabinet.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/hingecabinet.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/kettle.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/kettle.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/kitchen.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/kitchen.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/burnerplate.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/burnerplate.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/burnerplate_mesh.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/burnerplate_mesh.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/cabinetbase.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/cabinetbase.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/cabinetdrawer.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/cabinetdrawer.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/cabinethandle.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/cabinethandle.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/countertop.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/countertop.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/faucet.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/faucet.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/handle2.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/handle2.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/hingecabinet.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/hingecabinet.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/hingedoor.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/hingedoor.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/hingehandle.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/hingehandle.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/hood.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/hood.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/kettle.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/kettle.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/kettlehandle.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/kettlehandle.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/knob.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/knob.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/lightswitch.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/lightswitch.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/lightswitchbase.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/lightswitchbase.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/micro.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/micro.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/microbutton.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/microbutton.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/microdoor.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/microdoor.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/microefeet.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/microefeet.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/microfeet.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/microfeet.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/microhandle.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/microhandle.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/microwindow.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/microwindow.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/oven.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/oven.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/ovenhandle.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/ovenhandle.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/oventop.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/oventop.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/ovenwindow.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/ovenwindow.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/slidecabinet.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/slidecabinet.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/slidedoor.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/slidedoor.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/stoverim.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/stoverim.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/tile.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/tile.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/meshes/wall.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/meshes/wall.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/microwave.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/microwave.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/oven.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/oven.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/kitchen/slidecabinet.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/kitchen/slidecabinet.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/adept_models/scenes/basic_scene.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/adept_models/scenes/basic_scene.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/kitchen_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/kitchen_envs.py -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/LICENSE -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/README.md -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/assets/actuator0.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/assets/actuator0.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/assets/actuator1.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/assets/actuator1.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/assets/assets.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/assets/assets.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/assets/basic_scene.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/assets/basic_scene.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/assets/chain0.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/assets/chain0.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/assets/chain0_overlay.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/assets/chain0_overlay.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/assets/chain1.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/assets/chain1.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/assets/teleop_actuator.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/assets/teleop_actuator.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/bi-franka_panda.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/bi-franka_panda.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/franka_panda.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/franka_panda.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/franka_panda_teleop.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/franka_panda_teleop.xml -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/collision/finger.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/collision/finger.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/collision/hand.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/collision/hand.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link0.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link0.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link1.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link1.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link2.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link2.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link3.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link3.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link4.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link4.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link5.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link5.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link6.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link6.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link7.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/collision/link7.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/visual/finger.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/visual/finger.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/visual/hand.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/visual/hand.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link0.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link0.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link1.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link1.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link2.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link2.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link3.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link3.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link4.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link4.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link5.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link5.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link6.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link6.stl -------------------------------------------------------------------------------- /D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link7.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/kitchen/third_party/franka/meshes/visual/link7.stl -------------------------------------------------------------------------------- /D4RL/d4rl/locomotion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/locomotion/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/locomotion/ant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/locomotion/ant.py -------------------------------------------------------------------------------- /D4RL/d4rl/locomotion/assets/ant.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/locomotion/assets/ant.xml -------------------------------------------------------------------------------- /D4RL/d4rl/locomotion/assets/point.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/locomotion/assets/point.xml -------------------------------------------------------------------------------- /D4RL/d4rl/locomotion/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/locomotion/common.py -------------------------------------------------------------------------------- /D4RL/d4rl/locomotion/generate_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/locomotion/generate_dataset.py -------------------------------------------------------------------------------- /D4RL/d4rl/locomotion/goal_reaching_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/locomotion/goal_reaching_env.py -------------------------------------------------------------------------------- /D4RL/d4rl/locomotion/maze_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/locomotion/maze_env.py -------------------------------------------------------------------------------- /D4RL/d4rl/locomotion/mujoco_goal_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/locomotion/mujoco_goal_env.py -------------------------------------------------------------------------------- /D4RL/d4rl/locomotion/point.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/locomotion/point.py -------------------------------------------------------------------------------- /D4RL/d4rl/locomotion/swimmer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/locomotion/swimmer.py -------------------------------------------------------------------------------- /D4RL/d4rl/locomotion/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/locomotion/wrappers.py -------------------------------------------------------------------------------- /D4RL/d4rl/offline_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/offline_env.py -------------------------------------------------------------------------------- /D4RL/d4rl/ope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/ope.py -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/pointmaze/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze/dynamic_mjc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/pointmaze/dynamic_mjc.py -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze/gridcraft/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze/gridcraft/grid_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/pointmaze/gridcraft/grid_env.py -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze/gridcraft/grid_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/pointmaze/gridcraft/grid_spec.py -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze/gridcraft/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/pointmaze/gridcraft/utils.py -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze/gridcraft/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/pointmaze/gridcraft/wrappers.py -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze/maze_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/pointmaze/maze_model.py -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze/q_iteration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/pointmaze/q_iteration.py -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze/shortest_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/pointmaze/shortest_paths.py -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze/waypoint_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/pointmaze/waypoint_controller.py -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze_bullet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/pointmaze_bullet/__init__.py -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze_bullet/bullet_maze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/pointmaze_bullet/bullet_maze.py -------------------------------------------------------------------------------- /D4RL/d4rl/pointmaze_bullet/bullet_robot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/pointmaze_bullet/bullet_robot.py -------------------------------------------------------------------------------- /D4RL/d4rl/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /D4RL/d4rl/utils/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/utils/dataset_utils.py -------------------------------------------------------------------------------- /D4RL/d4rl/utils/quatmath.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/utils/quatmath.py -------------------------------------------------------------------------------- /D4RL/d4rl/utils/visualize_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/utils/visualize_env.py -------------------------------------------------------------------------------- /D4RL/d4rl/utils/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/d4rl/utils/wrappers.py -------------------------------------------------------------------------------- /D4RL/scripts/check_antmaze_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/check_antmaze_datasets.py -------------------------------------------------------------------------------- /D4RL/scripts/check_bullet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/check_bullet.py -------------------------------------------------------------------------------- /D4RL/scripts/check_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/check_envs.py -------------------------------------------------------------------------------- /D4RL/scripts/check_mujoco_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/check_mujoco_datasets.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/flow_idm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/flow_idm.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/generate_ant_maze_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/generate_ant_maze_datasets.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/generate_kitchen_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/generate_kitchen_datasets.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/generate_maze2d_bullet_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/generate_maze2d_bullet_datasets.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/generate_maze2d_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/generate_maze2d_datasets.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/generate_minigrid_fourroom_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/generate_minigrid_fourroom_data.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/hand_dapg_combined.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/hand_dapg_combined.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/hand_dapg_demos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/hand_dapg_demos.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/hand_dapg_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/hand_dapg_jax.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/hand_dapg_policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/hand_dapg_policies.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/hand_dapg_random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/hand_dapg_random.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/mujoco/collect_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/mujoco/collect_data.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/mujoco/convert_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/mujoco/convert_buffer.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/mujoco/fix_qpos_qvel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/mujoco/fix_qpos_qvel.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/mujoco/stitch_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/mujoco/stitch_dataset.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/relabel_antmaze_rewards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/relabel_antmaze_rewards.py -------------------------------------------------------------------------------- /D4RL/scripts/generation/relabel_maze2d_rewards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/generation/relabel_maze2d_rewards.py -------------------------------------------------------------------------------- /D4RL/scripts/ope_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/ope_rollout.py -------------------------------------------------------------------------------- /D4RL/scripts/reference_scores/adroit_expert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/reference_scores/adroit_expert.py -------------------------------------------------------------------------------- /D4RL/scripts/reference_scores/carla_lane_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/reference_scores/carla_lane_controller.py -------------------------------------------------------------------------------- /D4RL/scripts/reference_scores/generate_ref_min_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/reference_scores/generate_ref_min_score.py -------------------------------------------------------------------------------- /D4RL/scripts/reference_scores/generate_ref_min_score.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/reference_scores/generate_ref_min_score.sh -------------------------------------------------------------------------------- /D4RL/scripts/reference_scores/maze2d_bullet_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/reference_scores/maze2d_bullet_controller.py -------------------------------------------------------------------------------- /D4RL/scripts/reference_scores/maze2d_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/reference_scores/maze2d_controller.py -------------------------------------------------------------------------------- /D4RL/scripts/reference_scores/minigrid_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/reference_scores/minigrid_controller.py -------------------------------------------------------------------------------- /D4RL/scripts/visualize_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/scripts/visualize_dataset.py -------------------------------------------------------------------------------- /D4RL/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/D4RL/setup.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/README.md -------------------------------------------------------------------------------- /locomotion/.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/.dockerignore -------------------------------------------------------------------------------- /locomotion/.gitattributes: -------------------------------------------------------------------------------- 1 | *.ipynb linguist-vendored 2 | -------------------------------------------------------------------------------- /locomotion/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/.gitignore -------------------------------------------------------------------------------- /locomotion/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/Dockerfile -------------------------------------------------------------------------------- /locomotion/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/LICENSE -------------------------------------------------------------------------------- /locomotion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/README.md -------------------------------------------------------------------------------- /locomotion/Singularity.def: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/Singularity.def -------------------------------------------------------------------------------- /locomotion/azure/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/azure/config.py -------------------------------------------------------------------------------- /locomotion/azure/download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/azure/download.sh -------------------------------------------------------------------------------- /locomotion/azure/files/10_nvidia.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/azure/files/10_nvidia.json -------------------------------------------------------------------------------- /locomotion/azure/files/Xdummy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/azure/files/Xdummy -------------------------------------------------------------------------------- /locomotion/azure/files/mjkey.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/azure/files/mjkey.txt -------------------------------------------------------------------------------- /locomotion/azure/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/azure/launch.py -------------------------------------------------------------------------------- /locomotion/azure/sync.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/azure/sync.sh -------------------------------------------------------------------------------- /locomotion/config/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/config/discriminator.py -------------------------------------------------------------------------------- /locomotion/config/gradient_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/config/gradient_matching.py -------------------------------------------------------------------------------- /locomotion/config/locomotion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/config/locomotion.py -------------------------------------------------------------------------------- /locomotion/diffuser/__init__.py: -------------------------------------------------------------------------------- 1 | from . import environments -------------------------------------------------------------------------------- /locomotion/diffuser/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/datasets/__init__.py -------------------------------------------------------------------------------- /locomotion/diffuser/datasets/buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/datasets/buffer.py -------------------------------------------------------------------------------- /locomotion/diffuser/datasets/d4rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/datasets/d4rl.py -------------------------------------------------------------------------------- /locomotion/diffuser/datasets/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/datasets/normalization.py -------------------------------------------------------------------------------- /locomotion/diffuser/datasets/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/datasets/preprocessing.py -------------------------------------------------------------------------------- /locomotion/diffuser/datasets/sequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/datasets/sequence.py -------------------------------------------------------------------------------- /locomotion/diffuser/environments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/environments/__init__.py -------------------------------------------------------------------------------- /locomotion/diffuser/environments/ant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/environments/ant.py -------------------------------------------------------------------------------- /locomotion/diffuser/environments/assets/ant.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/environments/assets/ant.xml -------------------------------------------------------------------------------- /locomotion/diffuser/environments/assets/half_cheetah.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/environments/assets/half_cheetah.xml -------------------------------------------------------------------------------- /locomotion/diffuser/environments/assets/hopper.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/environments/assets/hopper.xml -------------------------------------------------------------------------------- /locomotion/diffuser/environments/assets/walker2d.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/environments/assets/walker2d.xml -------------------------------------------------------------------------------- /locomotion/diffuser/environments/half_cheetah.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/environments/half_cheetah.py -------------------------------------------------------------------------------- /locomotion/diffuser/environments/hopper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/environments/hopper.py -------------------------------------------------------------------------------- /locomotion/diffuser/environments/registration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/environments/registration.py -------------------------------------------------------------------------------- /locomotion/diffuser/environments/walker2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/environments/walker2d.py -------------------------------------------------------------------------------- /locomotion/diffuser/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/models/__init__.py -------------------------------------------------------------------------------- /locomotion/diffuser/models/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/models/diffusion.py -------------------------------------------------------------------------------- /locomotion/diffuser/models/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/models/helpers.py -------------------------------------------------------------------------------- /locomotion/diffuser/models/temporal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/models/temporal.py -------------------------------------------------------------------------------- /locomotion/diffuser/rrf_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/rrf_diffusion/__init__.py -------------------------------------------------------------------------------- /locomotion/diffuser/rrf_diffusion/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/rrf_diffusion/discriminator.py -------------------------------------------------------------------------------- /locomotion/diffuser/rrf_diffusion/discriminator_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/rrf_diffusion/discriminator_trainer.py -------------------------------------------------------------------------------- /locomotion/diffuser/rrf_diffusion/gradient_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/rrf_diffusion/gradient_matching.py -------------------------------------------------------------------------------- /locomotion/diffuser/rrf_diffusion/gradient_matching_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/rrf_diffusion/gradient_matching_trainer.py -------------------------------------------------------------------------------- /locomotion/diffuser/rrf_diffusion/preprocess_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/rrf_diffusion/preprocess_dataset.py -------------------------------------------------------------------------------- /locomotion/diffuser/rrf_diffusion/train_discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/rrf_diffusion/train_discriminator.py -------------------------------------------------------------------------------- /locomotion/diffuser/rrf_diffusion/train_gradient_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/rrf_diffusion/train_gradient_matching.py -------------------------------------------------------------------------------- /locomotion/diffuser/sampling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/sampling/__init__.py -------------------------------------------------------------------------------- /locomotion/diffuser/sampling/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/sampling/functions.py -------------------------------------------------------------------------------- /locomotion/diffuser/sampling/guides.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/sampling/guides.py -------------------------------------------------------------------------------- /locomotion/diffuser/sampling/policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/sampling/policies.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/__init__.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/arrays.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/arrays.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/cloud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/cloud.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/colab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/colab.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/config.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/datasets.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/debugging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/debugging.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/git_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/git_utils.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/iql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/iql.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/logger.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/progress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/progress.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/pybullet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/pybullet_utils.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/rendering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/rendering.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/serialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/serialization.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/setup.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/timer.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/training.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/transformations.py -------------------------------------------------------------------------------- /locomotion/diffuser/utils/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/diffuser/utils/video.py -------------------------------------------------------------------------------- /locomotion/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/environment.yml -------------------------------------------------------------------------------- /locomotion/experiment_commands/main_results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/experiment_commands/main_results.json -------------------------------------------------------------------------------- /locomotion/original_diffuser_readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/original_diffuser_readme.md -------------------------------------------------------------------------------- /locomotion/plotting/locomotion_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/plotting/locomotion_scores.py -------------------------------------------------------------------------------- /locomotion/plotting/table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/plotting/table.py -------------------------------------------------------------------------------- /locomotion/quetzal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/quetzal.yaml -------------------------------------------------------------------------------- /locomotion/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/requirements.txt -------------------------------------------------------------------------------- /locomotion/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/run.py -------------------------------------------------------------------------------- /locomotion/scripts/plan_discriminator_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/scripts/plan_discriminator_parallel.py -------------------------------------------------------------------------------- /locomotion/scripts/plan_gradient_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/scripts/plan_gradient_matching.py -------------------------------------------------------------------------------- /locomotion/scripts/plan_gradient_matching_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/scripts/plan_gradient_matching_parallel.py -------------------------------------------------------------------------------- /locomotion/scripts/plan_guided.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/scripts/plan_guided.py -------------------------------------------------------------------------------- /locomotion/scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/scripts/train.py -------------------------------------------------------------------------------- /locomotion/scripts/train_values.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/scripts/train_values.py -------------------------------------------------------------------------------- /locomotion/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/setup.py -------------------------------------------------------------------------------- /locomotion/sweep_configs/ablations/different_seed_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/sweep_configs/ablations/different_seed_base.yaml -------------------------------------------------------------------------------- /locomotion/sweep_configs/final/final_rollouts.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/locomotion/sweep_configs/final/final_rollouts.yaml -------------------------------------------------------------------------------- /locomotion/tmp_diffusion_commands.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /maze2d/.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/.dockerignore -------------------------------------------------------------------------------- /maze2d/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/.gitignore -------------------------------------------------------------------------------- /maze2d/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/LICENSE -------------------------------------------------------------------------------- /maze2d/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/README.md -------------------------------------------------------------------------------- /maze2d/config/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/config/discriminator.py -------------------------------------------------------------------------------- /maze2d/config/gradient_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/config/gradient_matching.py -------------------------------------------------------------------------------- /maze2d/config/locomotion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/config/locomotion.py -------------------------------------------------------------------------------- /maze2d/config/maze2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/config/maze2d.py -------------------------------------------------------------------------------- /maze2d/dataset_generation/generate_commands_d4rl_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/dataset_generation/generate_commands_d4rl_datasets.py -------------------------------------------------------------------------------- /maze2d/dataset_generation/make_json.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/dataset_generation/make_json.ipynb -------------------------------------------------------------------------------- /maze2d/dataset_generation/maze2d-large-v1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/dataset_generation/maze2d-large-v1.json -------------------------------------------------------------------------------- /maze2d/dataset_generation/maze2d-medium-v1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/dataset_generation/maze2d-medium-v1.json -------------------------------------------------------------------------------- /maze2d/dataset_generation/maze2d-open-v0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/dataset_generation/maze2d-open-v0.json -------------------------------------------------------------------------------- /maze2d/dataset_generation/maze2d-umaze-v0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/dataset_generation/maze2d-umaze-v0.json -------------------------------------------------------------------------------- /maze2d/dataset_generation/preprocessing_commands.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/dataset_generation/preprocessing_commands.txt -------------------------------------------------------------------------------- /maze2d/diffuser/__init__.py: -------------------------------------------------------------------------------- 1 | from . import environments -------------------------------------------------------------------------------- /maze2d/diffuser/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/datasets/__init__.py -------------------------------------------------------------------------------- /maze2d/diffuser/datasets/buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/datasets/buffer.py -------------------------------------------------------------------------------- /maze2d/diffuser/datasets/d4rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/datasets/d4rl.py -------------------------------------------------------------------------------- /maze2d/diffuser/datasets/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/datasets/normalization.py -------------------------------------------------------------------------------- /maze2d/diffuser/datasets/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/datasets/preprocessing.py -------------------------------------------------------------------------------- /maze2d/diffuser/datasets/sequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/datasets/sequence.py -------------------------------------------------------------------------------- /maze2d/diffuser/environments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/environments/__init__.py -------------------------------------------------------------------------------- /maze2d/diffuser/environments/ant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/environments/ant.py -------------------------------------------------------------------------------- /maze2d/diffuser/environments/assets/ant.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/environments/assets/ant.xml -------------------------------------------------------------------------------- /maze2d/diffuser/environments/assets/half_cheetah.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/environments/assets/half_cheetah.xml -------------------------------------------------------------------------------- /maze2d/diffuser/environments/assets/hopper.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/environments/assets/hopper.xml -------------------------------------------------------------------------------- /maze2d/diffuser/environments/assets/walker2d.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/environments/assets/walker2d.xml -------------------------------------------------------------------------------- /maze2d/diffuser/environments/half_cheetah.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/environments/half_cheetah.py -------------------------------------------------------------------------------- /maze2d/diffuser/environments/hopper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/environments/hopper.py -------------------------------------------------------------------------------- /maze2d/diffuser/environments/registration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/environments/registration.py -------------------------------------------------------------------------------- /maze2d/diffuser/environments/walker2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/environments/walker2d.py -------------------------------------------------------------------------------- /maze2d/diffuser/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/models/__init__.py -------------------------------------------------------------------------------- /maze2d/diffuser/models/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/models/diffusion.py -------------------------------------------------------------------------------- /maze2d/diffuser/models/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/models/helpers.py -------------------------------------------------------------------------------- /maze2d/diffuser/models/temporal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/models/temporal.py -------------------------------------------------------------------------------- /maze2d/diffuser/rrf_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/rrf_diffusion/__init__.py -------------------------------------------------------------------------------- /maze2d/diffuser/rrf_diffusion/depict_reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/rrf_diffusion/depict_reward.py -------------------------------------------------------------------------------- /maze2d/diffuser/rrf_diffusion/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/rrf_diffusion/discriminator.py -------------------------------------------------------------------------------- /maze2d/diffuser/rrf_diffusion/discriminator_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/rrf_diffusion/discriminator_trainer.py -------------------------------------------------------------------------------- /maze2d/diffuser/rrf_diffusion/gradient_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/rrf_diffusion/gradient_matching.py -------------------------------------------------------------------------------- /maze2d/diffuser/rrf_diffusion/gradient_matching_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/rrf_diffusion/gradient_matching_trainer.py -------------------------------------------------------------------------------- /maze2d/diffuser/rrf_diffusion/preprocess_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/rrf_diffusion/preprocess_dataset.py -------------------------------------------------------------------------------- /maze2d/diffuser/rrf_diffusion/train_discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/rrf_diffusion/train_discriminator.py -------------------------------------------------------------------------------- /maze2d/diffuser/rrf_diffusion/train_gradient_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/rrf_diffusion/train_gradient_matching.py -------------------------------------------------------------------------------- /maze2d/diffuser/sampling/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/sampling/functions.py -------------------------------------------------------------------------------- /maze2d/diffuser/sampling/guides.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/sampling/guides.py -------------------------------------------------------------------------------- /maze2d/diffuser/sampling/policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/sampling/policies.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/__init__.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/arrays.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/arrays.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/cloud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/cloud.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/colab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/colab.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/config.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/datasets.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/debugging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/debugging.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/git_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/git_utils.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/iql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/iql.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/progress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/progress.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/pybullet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/pybullet_utils.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/rendering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/rendering.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/serialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/serialization.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/setup.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/timer.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/training.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/transformations.py -------------------------------------------------------------------------------- /maze2d/diffuser/utils/video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffuser/utils/video.py -------------------------------------------------------------------------------- /maze2d/diffusion_training_commands/diffusion_commands.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffusion_training_commands/diffusion_commands.txt -------------------------------------------------------------------------------- /maze2d/diffusion_training_commands/make_json.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffusion_training_commands/make_json.ipynb -------------------------------------------------------------------------------- /maze2d/diffusion_training_commands/maze2d-large-v1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffusion_training_commands/maze2d-large-v1.json -------------------------------------------------------------------------------- /maze2d/diffusion_training_commands/maze2d-medium-v1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffusion_training_commands/maze2d-medium-v1.json -------------------------------------------------------------------------------- /maze2d/diffusion_training_commands/maze2d-open-v0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffusion_training_commands/maze2d-open-v0.json -------------------------------------------------------------------------------- /maze2d/diffusion_training_commands/maze2d-umaze-v0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/diffusion_training_commands/maze2d-umaze-v0.json -------------------------------------------------------------------------------- /maze2d/download_extract_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/download_extract_dataset.sh -------------------------------------------------------------------------------- /maze2d/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/environment.yml -------------------------------------------------------------------------------- /maze2d/experiment_commands/maze2d-large-v1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/experiment_commands/maze2d-large-v1.json -------------------------------------------------------------------------------- /maze2d/experiment_commands/maze2d-medium-v1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/experiment_commands/maze2d-medium-v1.json -------------------------------------------------------------------------------- /maze2d/experiment_commands/maze2d-open-v0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/experiment_commands/maze2d-open-v0.json -------------------------------------------------------------------------------- /maze2d/experiment_commands/maze2d-umaze-v0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/experiment_commands/maze2d-umaze-v0.json -------------------------------------------------------------------------------- /maze2d/installation.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/installation.txt -------------------------------------------------------------------------------- /maze2d/original_diffuser_readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/original_diffuser_readme.md -------------------------------------------------------------------------------- /maze2d/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/requirements.txt -------------------------------------------------------------------------------- /maze2d/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/run.py -------------------------------------------------------------------------------- /maze2d/scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/scripts/train.py -------------------------------------------------------------------------------- /maze2d/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/setup.py -------------------------------------------------------------------------------- /maze2d/sweep_configs/all_maze_experiments_10m_v1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/sweep_configs/all_maze_experiments_10m_v1.yaml -------------------------------------------------------------------------------- /maze2d/sweep_configs/dataset_size_ablation_1m_10k_v1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/maze2d/sweep_configs/dataset_size_ablation_1m_10k_v1.yaml -------------------------------------------------------------------------------- /pics/architecture_figure.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/pics/architecture_figure.jpg -------------------------------------------------------------------------------- /safe-stable-diffusion/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/.gitignore -------------------------------------------------------------------------------- /safe-stable-diffusion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/ISSUE_TEMPLATE/bug-report.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/ISSUE_TEMPLATE/bug-report.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/ISSUE_TEMPLATE/feedback.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/ISSUE_TEMPLATE/feedback.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/actions/setup-miniconda/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/actions/setup-miniconda/action.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/workflows/build_docker_images.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/workflows/build_docker_images.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/workflows/build_documentation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/workflows/build_documentation.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/workflows/delete_doc_comment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/workflows/delete_doc_comment.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/workflows/nightly_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/workflows/nightly_tests.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/workflows/pr_quality.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/workflows/pr_quality.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/workflows/pr_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/workflows/pr_tests.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/workflows/push_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/workflows/push_tests.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/workflows/push_tests_fast.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/workflows/push_tests_fast.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/workflows/stale.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/workflows/stale.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.github/workflows/typos.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.github/workflows/typos.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/.gitignore -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/CITATION.cff -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/CONTRIBUTING.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/LICENSE -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/MANIFEST.in -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/Makefile -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/PHILOSOPHY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/PHILOSOPHY.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/_typos.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/_typos.toml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docker/diffusers-flax-cpu/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docker/diffusers-flax-cpu/Dockerfile -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docker/diffusers-flax-tpu/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docker/diffusers-flax-tpu/Dockerfile -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docker/diffusers-onnxruntime-cpu/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docker/diffusers-onnxruntime-cpu/Dockerfile -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docker/diffusers-pytorch-cpu/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docker/diffusers-pytorch-cpu/Dockerfile -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docker/diffusers-pytorch-cuda/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docker/diffusers-pytorch-cuda/Dockerfile -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/TRANSLATING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/TRANSLATING.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/_config.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/_toctree.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/_toctree.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/configuration.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/configuration.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/diffusion_pipeline.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/diffusion_pipeline.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/experimental/rl.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/experimental/rl.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/loaders.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/loaders.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/logging.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/logging.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/models.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/models.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/outputs.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/outputs.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/audioldm.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/audioldm.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/ddim.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/ddim.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/ddpm.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/ddpm.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/dit.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/dit.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/if.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/if.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/overview.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/overview.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/pndm.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/pndm.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/repaint.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/repaint.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/unclip.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/pipelines/unclip.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/ddim.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/ddim.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/ddpm.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/ddpm.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/deis.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/deis.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/euler.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/euler.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/heun.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/heun.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/ipndm.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/ipndm.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/overview.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/overview.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/pndm.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/pndm.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/repaint.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/repaint.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/unipc.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/api/schedulers/unipc.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/conceptual/contribution.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/conceptual/contribution.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/conceptual/evaluation.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/conceptual/evaluation.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/conceptual/philosophy.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/conceptual/philosophy.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/imgs/access_request.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/imgs/access_request.png -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/imgs/diffusers_library.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/imgs/diffusers_library.jpg -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/index.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/index.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/installation.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/installation.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/optimization/coreml.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/optimization/coreml.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/optimization/fp16.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/optimization/fp16.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/optimization/habana.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/optimization/habana.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/optimization/mps.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/optimization/mps.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/optimization/onnx.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/optimization/onnx.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/optimization/tome.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/optimization/tome.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/optimization/torch2.0.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/optimization/torch2.0.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/optimization/xformers.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/optimization/xformers.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/quicktour.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/quicktour.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/stable_diffusion.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/stable_diffusion.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/training/controlnet.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/training/controlnet.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/training/dreambooth.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/training/dreambooth.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/training/lora.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/training/lora.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/training/overview.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/training/overview.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/training/text2image.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/training/text2image.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/using-diffusers/audio.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/using-diffusers/audio.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/en/using-diffusers/rl.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/en/using-diffusers/rl.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/ko/_toctree.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/ko/_toctree.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/ko/in_translation.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/ko/in_translation.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/ko/index.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/ko/index.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/ko/installation.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/ko/installation.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/ko/quicktour.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/ko/quicktour.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/zh/_toctree.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/zh/_toctree.yml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/zh/index.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/zh/index.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/zh/installation.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/zh/installation.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/docs/source/zh/quicktour.mdx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/docs/source/zh/quicktour.mdx -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/community/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/community/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/community/bit_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/community/bit_diffusion.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/community/checkpoint_merger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/community/checkpoint_merger.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/community/img2img_inpainting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/community/img2img_inpainting.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/community/magic_mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/community/magic_mix.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/community/one_step_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/community/one_step_unet.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/community/stable_unclip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/community/stable_unclip.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/community/text_inpainting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/community/text_inpainting.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/community/tiled_upscaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/community/tiled_upscaling.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/conftest.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/controlnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/controlnet/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/controlnet/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/controlnet/requirements.txt -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/controlnet/train_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/controlnet/train_controlnet.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/custom_diffusion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/custom_diffusion/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/custom_diffusion/retrieve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/custom_diffusion/retrieve.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/dreambooth/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/dreambooth/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/dreambooth/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/dreambooth/requirements.txt -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/dreambooth/train_dreambooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/dreambooth/train_dreambooth.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/inference/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/inference/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/inference/image_to_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/inference/image_to_image.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/inference/inpainting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/inference/inpainting.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/instruct_pix2pix/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/instruct_pix2pix/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/research_projects/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/research_projects/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/research_projects/onnxruntime/unconditional_image_generation/requirements.txt: -------------------------------------------------------------------------------- 1 | accelerate>=0.16.0 2 | torchvision 3 | datasets 4 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/rl/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/rl/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/rl/run_diffuser_locomotion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/rl/run_diffuser_locomotion.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/test_examples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/test_examples.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/text_to_image/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/text_to_image/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/text_to_image/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/text_to_image/requirements.txt -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/textual_inversion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/examples/textual_inversion/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/examples/unconditional_image_generation/requirements.txt: -------------------------------------------------------------------------------- 1 | accelerate>=0.16.0 2 | torchvision 3 | datasets 4 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/pyproject.toml -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/scripts/conversion_ldm_uncond.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/scripts/conversion_ldm_uncond.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/scripts/convert_dit_to_diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/scripts/convert_dit_to_diffusers.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/scripts/convert_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/scripts/convert_if.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/scripts/convert_vae_diff_to_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/scripts/convert_vae_diff_to_onnx.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/scripts/convert_vae_pt_to_diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/scripts/convert_vae_pt_to_diffusers.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/scripts/generate_logits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/scripts/generate_logits.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/setup.cfg -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/setup.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/__init__.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/commands/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/commands/__init__.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/commands/diffusers_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/commands/diffusers_cli.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/commands/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/commands/env.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/configuration_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/configuration_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/experimental/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/experimental/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/experimental/__init__.py: -------------------------------------------------------------------------------- 1 | from .rl import ValueGuidedRLPipeline 2 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/image_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/image_processor.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/loaders.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/__init__.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/attention.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/attention_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/attention_flax.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/autoencoder_kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/autoencoder_kl.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/controlnet.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/controlnet_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/controlnet_flax.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/cross_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/cross_attention.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/embeddings.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/embeddings_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/embeddings_flax.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/modeling_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/resnet.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/resnet_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/resnet_flax.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/transformer_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/transformer_2d.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/unet_1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/unet_1d.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/unet_1d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/unet_1d_blocks.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/unet_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/unet_2d.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/unet_2d_blocks.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/unet_3d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/unet_3d_blocks.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/vae.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/vae_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/vae_flax.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/models/vq_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/models/vq_model.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/optimization.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/pipeline_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/pipeline_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/pipelines/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/pipelines/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/pipelines/__init__.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/pipelines/ddim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/pipelines/ddim/__init__.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/pipelines/ddpm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/pipelines/ddpm/__init__.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/pipelines/dit/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/pipelines/dit/__init__.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/pipelines/onnx_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/pipelines/onnx_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/pipelines/pndm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/pipelines/pndm/__init__.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/schedulers/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/schedulers/README.md -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/schedulers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/schedulers/__init__.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/training_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/training_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/utils/__init__.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/utils/accelerate_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/utils/accelerate_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/utils/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/utils/constants.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/utils/deprecation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/utils/deprecation_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/utils/doc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/utils/doc_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/utils/dummy_pt_objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/utils/dummy_pt_objects.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/utils/hub_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/utils/hub_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/utils/import_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/utils/import_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/utils/logging.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/utils/outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/utils/outputs.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/utils/pil_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/utils/pil_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/utils/testing_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/utils/testing_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/src/diffusers/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/src/diffusers/utils/torch_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/conftest.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/fixtures/elise_format0.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/fixtures/elise_format0.mid -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/models/test_attention_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/models/test_attention_processor.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/models/test_layers_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/models/test_layers_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/models/test_lora_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/models/test_lora_layers.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/models/test_modeling_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/models/test_modeling_common.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/models/test_models_unet_1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/models/test_models_unet_1d.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/models/test_models_unet_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/models/test_models_unet_2d.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/models/test_models_unet_2d_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/models/test_models_unet_2d_flax.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/models/test_models_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/models/test_models_vae.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/models/test_models_vae_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/models/test_models_vae_flax.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/models/test_models_vq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/models/test_models_vq.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/models/test_unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/models/test_unet_2d_blocks.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/models/test_unet_blocks_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/models/test_unet_blocks_common.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/others/test_check_copies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/others/test_check_copies.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/others/test_check_dummies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/others/test_check_dummies.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/others/test_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/others/test_config.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/others/test_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/others/test_ema.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/others/test_hub_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/others/test_hub_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/others/test_image_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/others/test_image_processor.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/others/test_outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/others/test_outputs.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/others/test_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/others/test_training.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/others/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/others/test_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/altdiffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/audio_diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/audioldm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/dance_diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/ddim/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/ddim/test_ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/pipelines/ddim/test_ddim.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/ddpm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/ddpm/test_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/pipelines/ddpm/test_ddpm.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/deepfloyd_if/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/pipelines/deepfloyd_if/__init__.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/deepfloyd_if/test_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/pipelines/deepfloyd_if/test_if.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/dit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/dit/test_dit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/pipelines/dit/test_dit.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/karras_ve/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/latent_diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/paint_by_example/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/pipeline_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/pipelines/pipeline_params.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/pndm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/pndm/test_pndm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/pipelines/pndm/test_pndm.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/repaint/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/repaint/test_repaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/pipelines/repaint/test_repaint.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/score_sde_ve/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/semantic_stable_diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/spectrogram_diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/stable_diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/stable_diffusion_2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/stable_diffusion_safe/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/stable_unclip/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/test_pipeline_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/pipelines/test_pipeline_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/test_pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/pipelines/test_pipelines.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/test_pipelines_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/pipelines/test_pipelines_common.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/test_pipelines_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/pipelines/test_pipelines_flax.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/text_to_video/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/unclip/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/unclip/test_unclip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/pipelines/unclip/test_unclip.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/versatile_diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/pipelines/vq_diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/schedulers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_ddim.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_ddpm.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_deis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_deis.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_euler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_euler.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_flax.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_heun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_heun.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_ipndm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_ipndm.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_lms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_lms.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_pndm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_pndm.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_unipc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/schedulers/test_scheduler_unipc.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/tests/schedulers/test_schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/tests/schedulers/test_schedulers.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/utils/check_config_docstrings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/utils/check_config_docstrings.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/utils/check_copies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/utils/check_copies.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/utils/check_doc_toc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/utils/check_doc_toc.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/utils/check_dummies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/utils/check_dummies.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/utils/check_inits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/utils/check_inits.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/utils/check_repo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/utils/check_repo.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/utils/check_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/utils/check_table.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/utils/custom_init_isort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/utils/custom_init_isort.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/utils/get_modified_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/utils/get_modified_files.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/utils/overwrite_expected_slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/utils/overwrite_expected_slice.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/utils/print_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/utils/print_env.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/utils/release.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/utils/release.py -------------------------------------------------------------------------------- /safe-stable-diffusion/diffusers/utils/stale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/diffusers/utils/stale.py -------------------------------------------------------------------------------- /safe-stable-diffusion/download_extract_safe_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/download_extract_safe_data.sh -------------------------------------------------------------------------------- /safe-stable-diffusion/notebooks/rank_and_decode.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/notebooks/rank_and_decode.ipynb -------------------------------------------------------------------------------- /safe-stable-diffusion/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/requirements.txt -------------------------------------------------------------------------------- /safe-stable-diffusion/rrf_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/rrf_diffusion/__init__.py -------------------------------------------------------------------------------- /safe-stable-diffusion/rrf_diffusion/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/rrf_diffusion/dataset.py -------------------------------------------------------------------------------- /safe-stable-diffusion/rrf_diffusion/gradient_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/rrf_diffusion/gradient_matching.py -------------------------------------------------------------------------------- /safe-stable-diffusion/rrf_diffusion/gradient_matching_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/rrf_diffusion/gradient_matching_trainer.py -------------------------------------------------------------------------------- /safe-stable-diffusion/rrf_diffusion/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/rrf_diffusion/models.py -------------------------------------------------------------------------------- /safe-stable-diffusion/rrf_diffusion/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/rrf_diffusion/utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/rrf_diffusion/utils/git_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/rrf_diffusion/utils/git_utils.py -------------------------------------------------------------------------------- /safe-stable-diffusion/rrf_diffusion/utils/serialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/rrf_diffusion/utils/serialization.py -------------------------------------------------------------------------------- /safe-stable-diffusion/rrf_diffusion/utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/rrf_diffusion/utils/setup.py -------------------------------------------------------------------------------- /safe-stable-diffusion/scripts/train_gradient_discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/scripts/train_gradient_discriminator.py -------------------------------------------------------------------------------- /safe-stable-diffusion/scripts/train_gradient_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/scripts/train_gradient_matching.py -------------------------------------------------------------------------------- /safe-stable-diffusion/utils/generate_latents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/utils/generate_latents.py -------------------------------------------------------------------------------- /safe-stable-diffusion/utils/merge_latents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FelipeNuti/diffusion-relative-rewards/HEAD/safe-stable-diffusion/utils/merge_latents.py --------------------------------------------------------------------------------