a

main.py

@@ -87,10 +87,13 @@ def generate(request: GenerationRequest):
             return ErrorOutput(error=output[1])
         
         if request.advanced:
+            if request.n_samples > 1:
+                return ErrorOutput(error="advanced mode does not support n_samples > 1")
+                
             images = model.sample_two_stages(request)
         else:
             images = model.sample(request)
-            
+
         images_encoded = []
         for x in range(len(images)):
             image = simplejpeg.encode_jpeg(images[x], quality=95)