add all files

README.md

@@ -18,6 +18,6 @@ python run.py --model_id stabilityai/stable-diffusion-xl-base-1.0
 python run.py --model_id PixArt-alpha/PixArt-XL-2-512x512 
 python run.py --model_id PixArt-alpha/PixArt-XL-2-1024-MS 
 python run.py --model_id playgroundai/playground-v2-1024px-aesthetic
-python run.py --model_id kandinsky-community/kandinsky-3
+python run.py --model_id kandinsky-community/kandinsky-3 --variant fp16
 python run.py --model_id Lykon/dreamshaper-8
 ```

run.py

@@ -29,6 +29,7 @@ def main(
     prompt_path="assets/ViLG-300.csv",
     save_path=None,
     dtype='fp16',
+    variant=None,
 ):
     if save_path is None:
         save_path = os.path.join('saved', model_id.replace('/', '_'))
@@ -37,6 +38,7 @@ def main(
     prompts = load_prompts(prompt_path)
     pipeline = AutoPipelineForText2Image.from_pretrained(
         model_id, 
+        variant=variant, 
         torch_dtype=torch.float32 if dtype == 'fp32' else torch.float16
     )
     pipeline.to(device='cuda')