Spaces:
Running
on
Zero
Running
on
Zero
alfredplpl
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -14,15 +14,15 @@ import requests
|
|
14 |
|
15 |
model_id = 'aipicasso/cool-japan-diffusion-2-1-2'
|
16 |
|
17 |
-
scheduler = EulerAncestralDiscreteScheduler
|
18 |
-
feature_extractor = CLIPImageProcessor
|
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,
|