启动gradio时,启动脚本本身报错。环境为WSL2下的Debian,物理机(Windows 11)存储
$ python gradio_app.py --model_path . --offload --quantized
...
(此时模型已经加载到GPU上)
Traceback (most recent call last):
File "/mnt/h/3jindustry/step1xedit/gradio_app.py", line 535, in
demo = create_demo(args)
^^^^^^^^^^^^^^^^^
File "/mnt/h/3jindustry/step1xedit/gradio_app.py", line 487, in create_demo
inference_func = prepare_infer_func(args)
^^^^^^^^^^^^^^^^^^^^^^^^
File "/mnt/h/3jindustry/step1xedit/gradio_app.py", line 450, in prepare_infer_func
image_edit = ImageGenerator(
^^^^^^^^^^^^^^^
File "/mnt/h/3jindustry/step1xedit/gradio_app.py", line 131, in init
self.ae, self.dit, self.llm_encoder = load_models(
^^^^^^^^^^^^
File "/mnt/h/3jindustry/step1xedit/gradio_app.py", line 80, in load_models
step1x_params = Step1XParams(
^^^^^^^^^^^^^
TypeError: Step1XParams.init() missing 2 required positional arguments: 'mode' and 'version'