d8ahazard / sd_dreambooth_extension

Other
1.85k stars 281 forks source link

[Bug]: can't create new model from ckpt #1379

Closed lilyzlt closed 8 months ago

lilyzlt commented 8 months ago

Is there an existing issue for this?

What happened?

create new model from old models(.ckpt),get error image

Command Line Arguments

python launch.py --enable-insecure-extension-access --api --deepdanbooru --listen --port 52369

Console logs

result = extract_checkpoint(new_model_name=new_model_name,
  File "/data/stable-diffusion-webui/extensions/sd_dreambooth_extension/dreambooth/sd_to_diff.py", line 151, in extract_checkpoint
    original_config_file = get_config_file(train_unfrozen, model_type)
  File "/data/stable-diffusion-webui/extensions/sd_dreambooth_extension/dreambooth/sd_to_diff.py", line 82, in get_config_file
    model_version_name = model_versions[model_type]
KeyError: ''

Traceback (most recent call last):
  File "/data/miniconda3/envs/env-novelai/lib/python3.10/site-packages/gradio/routes.py", line 488, in run_predict
    output = await app.get_blocks().process_api(
  File "/data/miniconda3/envs/env-novelai/lib/python3.10/site-packages/gradio/blocks.py", line 1434, in process_api
    data = self.postprocess_data(fn_index, result["prediction"], state)
  File "/data/miniconda3/envs/env-novelai/lib/python3.10/site-packages/gradio/blocks.py", line 1297, in postprocess_data
    self.validate_outputs(fn_index, predictions)  # type: ignore
  File "/data/miniconda3/envs/env-novelai/lib/python3.10/site-packages/gradio/blocks.py", line 1272, in validate_outputs
    raise ValueError(
ValueError: An event handler (f) didn't receive enough output values (needed: 10, received: 3).
Wanted outputs:
    [dropdown, html, html, html, html, html, html, html, slider, html]
Received outputs:
    [None, "", "<div class='error'>KeyError: &#x27;&#x27;</div>"]
Error completing request
Arguments: ('vtest', 'novelAI.ckpt [a7529df023]', '', False, '', '', False, False, '') {}

Additional information

No response

mostafahazim commented 8 months ago

Facing the same issue

d8ahazard commented 8 months ago

Please try the dev branch, this should be resolved.

On Wed, Nov 8, 2023, 7:49 AM Hazem @.***> wrote:

Facing the same issue

— Reply to this email directly, view it on GitHub https://github.com/d8ahazard/sd_dreambooth_extension/issues/1379#issuecomment-1801927357, or unsubscribe https://github.com/notifications/unsubscribe-auth/AAMO4NF6OJUCLHXV2D6WEWLYDOEVDAVCNFSM6AAAAAA62SUZUCVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMYTQMBRHEZDOMZVG4 . You are receiving this because you are subscribed to this thread.Message ID: @.***>

mostafahazim commented 8 months ago

@d8ahazard thanks working fine on dev branch

d8ahazard commented 8 months ago

Closed as changes are added to main.

lilyzlt commented 8 months ago

still the same image

image