[Bug]: Can't create a model from huggingface hub

[lukaville]

Is there an existing issue for this?

• [X] I have searched the existing issues and checked the recent builds/commits of both this extension and the webui

What happened?

When creating a model like this:

It fails to create it.

Steps to reproduce the problem

1. Set huggingface model url/token
2. Cilck create button

Commit and libraries

Initializing Dreambooth
Dreambooth revision: d8e01e1fd13dd4b6afadbb0382875f905ee5f429
Installing bitsandbytes
Bitsandbytes 0.41.1 installation failed
Some features such as 8bit optimizers will be unavailable
Install manually with
'python -m pip install bitsandbytes==0.41.1  --prefer-binary --force-install'
[!] xformers NOT installed.
[+] torch version 2.0.1+cu118 installed.
[+] torchvision version 0.15.2+cu118 installed.
[+] accelerate version 0.23.0 installed.
[+] diffusers version 0.21.4 installed.
[+] transformers version 4.32.1 installed.
[+] bitsandbytes version 0.41.2.post2 installed.

Command Line Arguments

./webui.sh --listen --enable-insecure-extension-access

Console logs

Extracting config from /root/stable-diffusion-webui/extensions/sd_dreambooth_extension/dreambooth/../configs/v1-training-default.yaml
Extracting checkpoint from /root/stable-diffusion-webui/models/
Something went wrong, removing model directory
Traceback (most recent call last):
  File "/root/stable-diffusion-webui/extensions/sd_dreambooth_extension/dreambooth/sd_to_diff.py", line 198, in extract_checkpoint
    pipe = StableDiffusionPipeline.from_pretrained(
  File "/root/stable-diffusion-webui/venv/lib/python3.10/site-packages/diffusers/pipelines/pipeline_utils.py", line 960, in from_pretrained
    config_dict = cls.load_config(cached_folder)
  File "/root/stable-diffusion-webui/venv/lib/python3.10/site-packages/diffusers/configuration_utils.py", line 364, in load_config
    raise EnvironmentError(
OSError: Error no file named model_index.json found in directory /root/stable-diffusion-webui/models/.
Couldn't find /root/stable-diffusion-webui/models/dreambooth/iris_dreambooth/working/unet
Unable to extract checkpoint!
Duration: 00:00:01

Additional information

It doesn't look like the parameters new_model_url/new_model_token are used in create_model function? https://github.com/d8ahazard/sd_dreambooth_extension/blob/e9f8e4ee3548d76dffef03447783e6f0f1ebd707/dreambooth/ui_functions.py#L940

[github-actions[bot]]

This issue is stale because it has been open 5 days with no activity. Remove stale label or comment or this will be closed in 5 days