alfredplpl commited on
Commit
f811c83
1 Parent(s): 76653a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,6 +1,6 @@
1
  # Thank AK. https://huggingface.co/spaces/akhaliq/cool-japan-diffusion-2-1-0/blob/main/app.py
2
  from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, EulerAncestralDiscreteScheduler,StableDiffusionLatentUpscalePipeline
3
- from transformers import CLIPFeatureExtractor
4
  import gradio as gr
5
  import torch
6
  from PIL import Image
@@ -8,8 +8,8 @@ import random
8
 
9
  model_id = 'aipicasso/cool-japan-diffusion-2-1-2'
10
 
11
- scheduler = EulerAncestralDiscreteScheduler() # .from_pretrained(model_id) # , subfolder="scheduler"
12
- feature_extractor = CLIPFeatureExtractor.from_pretrained(model_id)
13
 
14
  pipe = StableDiffusionPipeline.from_pretrained(
15
  model_id,
 
1
  # Thank AK. https://huggingface.co/spaces/akhaliq/cool-japan-diffusion-2-1-0/blob/main/app.py
2
  from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, EulerAncestralDiscreteScheduler,StableDiffusionLatentUpscalePipeline
3
+ from transformers import CLIPImageProcessor
4
  import gradio as gr
5
  import torch
6
  from PIL import Image
 
8
 
9
  model_id = 'aipicasso/cool-japan-diffusion-2-1-2'
10
 
11
+ scheduler = EulerAncestralDiscreteScheduler(prediction_type="v_prediction") # .from_pretrained(model_id) # , subfolder="scheduler"
12
+ feature_extractor = CLIPImageProcessor.from_pretrained(model_id)
13
 
14
  pipe = StableDiffusionPipeline.from_pretrained(
15
  model_id,