update cuda version in replicate docker image

2022-07-05 15:07:32 -04:00
parent 3aa67b4f5d
commit cc0450337c
4 changed files with 5 additions and 6 deletions
@@ -1,7 +1,6 @@
 import argparse
 import os
 from PIL import Image
-from matplotlib.pyplot import grid
 from min_dalle import MinDalle