jadechoghari
/

mar

jadechoghari commited on Sep 11

Commit

4bd156b

•

1 Parent(s): 99cd496

Update pipeline.py

Files changed (1) hide show

pipeline.py CHANGED Viewed

@@ -95,7 +95,7 @@ class MARModel(DiffusionPipeline):
         cfg_scale = kwargs.get("cfg_scale", 4)
         cfg_schedule = kwargs.get("cfg_schedule", "constant")
         temperature = kwargs.get("temperature", 1.0)
-        class_labels = kwargs.get("class_labels", 207, 360, 388, 113, 355, 980, 323, 979)
         # generate the tokens and images
         with torch.cuda.amp.autocast():

         cfg_scale = kwargs.get("cfg_scale", 4)
         cfg_schedule = kwargs.get("cfg_schedule", "constant")
         temperature = kwargs.get("temperature", 1.0)
+        class_labels = kwargs.get("class_labels", [207, 360, 388, 113, 355, 980, 323, 979])
         # generate the tokens and images
         with torch.cuda.amp.autocast():