benchmark
560a4ee5 - Add instructpix2pix model (#1451)

Commit
2 years ago
Add instructpix2pix model (#1451) Summary: I place this model in the `canary_model` directory for the following reasons: 1. The model only has inference test, no train test 2. The model installation is very slow, taking ~5 mins and is downloading large model checkpoints (~10GB) 3. The inference test is slow (~4 seconds) and has noisy outputs It needs further work to make it suitable to the core model set. Fixes https://github.com/pytorch/benchmark/issues/1444 Pull Request resolved: https://github.com/pytorch/benchmark/pull/1451 Test Plan: ``` $ python run.py diffuser_instruct_pix2pix -d cuda Fetching 15 files: 100%|███████████████| 15/15 [00:00<00:00, 27449.63it/s] Running eval method from diffuser_instruct_pix2pix on cuda in eager mode with input batch size 1. ███████████████████████████████████████████| 100/100 [00:04<00:00, 23.96it/s] GPU Time: 4123.949 milliseconds CPU Total Wall Time: 4123.980 milliseconds GPU 0 Peak Memory: 4.2791 GB CPU Peak Memory: 3.7969 GB ``` Reviewed By: msaroufim Differential Revision: D43842040 Pulled By: xuzhao9 fbshipit-source-id: adc21408b3562076c471c720863d3f135e58c7f8
Author
Parents
Loading