alfredplpl commited on
Commit
c557830
1 Parent(s): 2c7d2b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,15 +14,15 @@ import requests
14
 
15
  model_id = 'aipicasso/cool-japan-diffusion-2-1-2'
16
 
17
- scheduler = EulerAncestralDiscreteScheduler(prediction_type="v_prediction") # .from_pretrained(model_id) # , subfolder="scheduler"
18
- feature_extractor = CLIPImageProcessor() # .from_pretrained(model_id)
19
 
20
  pipe = StableDiffusionPipeline.from_pretrained(
21
  model_id,
22
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
23
  scheduler=scheduler)
24
  pipe.enable_xformers_memory_efficient_attention()
25
- pipe.enable_freeu(s1=0.9, s2=0.2, b1=1.4, b2=1.6)
26
  # b1: 1.4, b2: 1.6, s1: 0.9, s2: 0.2
27
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
28
  model_id,
 
14
 
15
  model_id = 'aipicasso/cool-japan-diffusion-2-1-2'
16
 
17
+ scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id) # , subfolder="scheduler"
18
+ feature_extractor = CLIPImageProcessor.from_pretrained(model_id)
19
 
20
  pipe = StableDiffusionPipeline.from_pretrained(
21
  model_id,
22
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
23
  scheduler=scheduler)
24
  pipe.enable_xformers_memory_efficient_attention()
25
+ #pipe.enable_freeu(s1=0.9, s2=0.2, b1=1.4, b2=1.6)
26
  # b1: 1.4, b2: 1.6, s1: 0.9, s2: 0.2
27
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
28
  model_id,