diff --git a/server.py b/server.py
index ec6deb1..c66b4f4 100644
--- a/server.py
+++ b/server.py
@@ -239,10 +239,10 @@ def main_dev(opt):
             # 设置参数
             if 'prompt' in item:
                 opt.prompt = item['prompt']    # 描述
-            if 'n_samples' in item:
+            if 'number' in item:
                 opt.n_samples = item['number']  # 列数
-            if 'n_rows' in item:
-                opt.n_rows = item['n_rows']    # 行数
+            #if 'n_rows' in item:
+            #    opt.n_rows = item['n_rows']    # 行数
             if 'scale' in item:
                 opt.scale = item['scale']     # 比例
 
@@ -276,7 +276,7 @@ def main_dev(opt):
             wm_encoder.set_watermark('bytes', wm.encode('utf-8'))
             # x
             batch_size = opt.n_samples
-            n_rows = opt.n_rows if opt.n_rows > 0 else batch_size
+            #n_rows = opt.n_rows if opt.n_rows > 0 else batch_size
             if not opt.from_file:
                 prompt = opt.prompt
                 assert prompt is not None