diff --git a/README.md b/README.md
index 70a143a..7542f2f 100644
--- a/README.md
+++ b/README.md
@@ -56,7 +56,8 @@ use Imactool\HyperfStableDiffusion\Replicate;
                 ->highlyDetailed()
                 ->dramaticLighting()
                 ->octaneRender()
-        )->generate(3);
+        )->inputParams('num_outputs', 1)
+        ->generate(); //根据不同的模型来传递生成的张数
 ```
 
 ### 图片生成图片(Image to Image)
@@ -82,7 +83,8 @@ $result = Replicate::make()
     ->inputParams('num_inference_steps', 25)
     ->inputParams('guidance_scale', 7.5)
     ->inputParams('scheduler', 'EulerAncestralDiscrete')
-    ->generate(1);
+    ->inputParams('num_outputs', 1)
+    ->generate();
 ```
 
 
diff --git a/src/Replicate.php b/src/Replicate.php
index ea0b461..f9e3752 100644
--- a/src/Replicate.php
+++ b/src/Replicate.php
@@ -184,14 +184,14 @@ public function height(int $height)
             return $this;
         }
 
-        public function generate(int $numberOfImages)
+        public function generate()
         {
             assert($this->prompt !== null, 'You must provide a prompt');
-            assert($numberOfImages > 0, 'You must provide a number greater than 0');
+//            assert($numberOfImages > 0, 'You must provide a number greater than 0');
 
             $input = [
                 'prompt' => $this->prompt->toString(),
-                'num_outputs' => $numberOfImages,
+                //                'num_outputs' => $numberOfImages,
             ];
 
             $input = array_merge($input, $this->inputParams);