Feat: stable-diffusion-v2

[jagadeeshi2i]

Upgrade stable-diffusion model to v2

Signed-off-by: jagadeesh jagadeeshj@ideas2it.com

Description

Please read our CONTRIBUTING.md prior to creating your first pull request.

Please include a summary of the feature or issue being fixed. Please also include relevant motivation and context. List any dependencies that are required for this change.

Fixes #(issue)

Type of change

Please delete options that are not relevant.

• Bug fix (non-breaking change which fixes an issue)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• New feature (non-breaking change which adds functionality)
• This change requires a documentation update

Feature/Issue validation/testing

Please describe the Unit or Integration tests that you ran to verify your changes and relevant result summary. Provide instructions so it can be reproduced.
Please also list any relevant details for your test configuration.

• Test A
  Logs for Test A

• Test B
  Logs for Test B

Checklist:

• Did you have fun?
• Have you added tests that prove your fix is effective or that this feature works?
• Has code been commented, particularly in hard-to-understand areas?
• Have you made corresponding changes to the documentation?

[codecov]

Codecov Report

Merging #2009 (1e0e17d) into master (5cc8d0f) will not change coverage.
The diff coverage is n/a.

@@           Coverage Diff           @@
##           master    #2009   +/-   ##
=======================================
  Coverage   53.38%   53.38%           
=======================================
  Files          71       71           
  Lines        3224     3224           
  Branches       56       56           
=======================================
  Hits         1721     1721           
  Misses       1503     1503           

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

[maaquib]

$ python Download_model.py
Fetching 12 files: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 12/12 [00:00<00:00, 28055.55it/s]
The config attributes {'requires_safety_checker': False} were passed to StableDiffusionPipeline, but are not expected and will be ignored. Please verify your model_index.json configuration file.
The config attributes {'prediction_type': 'v_prediction', 'skip_prk_steps': True} were passed to DDIMScheduler, but are not expected and will be ignored. Please verify your scheduler_config.json configuration file.
Traceback (most recent call last):
  File "Download_model.py", line 6, in <module>
    pipeline = DiffusionPipeline.from_pretrained(
  File "/home/ubuntu/.local/lib/python3.8/site-packages/diffusers/pipeline_utils.py", line 452, in from_pretrained
    if class_name.startswith("Flax"):
AttributeError: 'NoneType' object has no attribute 'startswith'

$ pip freeze | grep diffusers
diffusers==0.6.0

@jagadeeshi2i Can you also add pip install -r requirements.txt to the readme as a step

stale