Update README.md

README.md

@@ -1,7 +1,7 @@
----
-library_name: diffusers
-license: mit
----
+---
+library_name: diffusers
+license: mit
+---
 
 # Autoregressive Image Generation without Vector Quantization
 
@@ -23,12 +23,13 @@ pipeline = DiffusionPipeline.from_pretrained("jadechoghari/mar", trust_remote_co
 
 # generate an image with the model
 generated_image = pipeline(
-    model_type="mar_base",  # choose from 'mar_base', 'mar_large', or 'mar_huge'
+    model_type="mar_huge",  # choose from 'mar_base', 'mar_large', or 'mar_huge'
     seed=42,                # set a seed for reproducibility
     num_ar_steps=64,        # number of autoregressive steps
     class_labels=[207, 360, 388],  # provide valid ImageNet class labels
     cfg_scale=4,            # classifier-free guidance scale
     output_dir="./images",   # directory to save generated images
+    cfg_schedule = "constant", # choose between 'constant' (suggested) and 'linear'
 )
 
 # display the generated image