wvangils commited on
Commit
3935025
1 Parent(s): d5c6181

Stable Diffusion fix + Bloom 560m

Browse files

Accelerated 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.

Files changed (1) hide show
  1. app.py +3 -3
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
- # 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")
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