Spaces:
Runtime error
Runtime error
Stable Diffusion fix + Bloom 560m
Browse filesAccelerated Gradio backend for Stable diffusion seems to be working again (at least in Colab). Also: select Bloom as default model since it has the best eval.
app.py
CHANGED
@@ -39,15 +39,15 @@ def get_image_input(title, given_input_style):
|
|
39 |
return image_input, final_style
|
40 |
|
41 |
# Available models for generate lyrics pipeline
|
42 |
-
checkpoint = 'wvangils/GPT-Medium-Beatles-Lyrics-finetuned-newlyrics'
|
43 |
# checkpoint = 'wvangils/GPT-Neo-125m-Beatles-Lyrics-finetuned-newlyrics'
|
44 |
-
|
45 |
|
46 |
# Setup all the pipelines we need
|
47 |
title_generator = pipeline('summarization', model='czearing/story-to-title')
|
48 |
lyrics_generator = pipeline("text-generation", model=checkpoint)
|
49 |
# For the image generator we use stable diffusion from an existing HuggingFace space
|
50 |
-
stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
|
51 |
|
52 |
# Create 4 images for the given prompt and receive the first one
|
53 |
# This function uses an existing HuggingFace space where the number of created images cannot be modified
|
|
|
39 |
return image_input, final_style
|
40 |
|
41 |
# Available models for generate lyrics pipeline
|
42 |
+
# checkpoint = 'wvangils/GPT-Medium-Beatles-Lyrics-finetuned-newlyrics'
|
43 |
# checkpoint = 'wvangils/GPT-Neo-125m-Beatles-Lyrics-finetuned-newlyrics'
|
44 |
+
checkpoint = 'wvangils/BLOOM-560m-Beatles-Lyrics-finetuned'
|
45 |
|
46 |
# Setup all the pipelines we need
|
47 |
title_generator = pipeline('summarization', model='czearing/story-to-title')
|
48 |
lyrics_generator = pipeline("text-generation", model=checkpoint)
|
49 |
# For the image generator we use stable diffusion from an existing HuggingFace space
|
50 |
+
stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion-1")
|
51 |
|
52 |
# Create 4 images for the given prompt and receive the first one
|
53 |
# This function uses an existing HuggingFace space where the number of created images cannot be modified
|