Spaces:
Runtime error
Runtime error
app update
Browse files
app.py
CHANGED
|
@@ -38,6 +38,11 @@ vae = vae.to(torch_device)
|
|
| 38 |
text_encoder = text_encoder.to(torch_device)
|
| 39 |
unet = unet.to(torch_device)
|
| 40 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
seed_values = [8,16,50,80,128]
|
| 42 |
height = 512 # default height of Stable Diffusion
|
| 43 |
width = 512 # default width of Stable Diffusion
|
|
@@ -75,9 +80,6 @@ def get_output_embeds(input_embeddings):
|
|
| 75 |
# And now they're ready!
|
| 76 |
return output
|
| 77 |
|
| 78 |
-
style_files = ['stable_diffusion/learned_embeddings/arcane-style-jv.bin', 'stable_diffusion/learned_embeddings/birb-style.bin',
|
| 79 |
-
'stable_diffusion/learned_embeddings/dr-strange.bin', 'stable_diffusion/learned_embeddings/midjourney-style.bin',
|
| 80 |
-
'stable_diffusion/learned_embeddings/oil_style.bin']
|
| 81 |
|
| 82 |
def get_style_embeddings(style_file):
|
| 83 |
style_embed = torch.load(style_file)
|
|
|
|
| 38 |
text_encoder = text_encoder.to(torch_device)
|
| 39 |
unet = unet.to(torch_device)
|
| 40 |
|
| 41 |
+
style_files = ['stable_diffusion/learned_embeddings/arcane-style-jv.bin', 'stable_diffusion/learned_embeddings/birb-style.bin',
|
| 42 |
+
'stable_diffusion/learned_embeddings/dr-strange.bin', 'stable_diffusion/learned_embeddings/midjourney-style.bin',
|
| 43 |
+
'stable_diffusion/learned_embeddings/oil_style.bin']
|
| 44 |
+
|
| 45 |
+
|
| 46 |
seed_values = [8,16,50,80,128]
|
| 47 |
height = 512 # default height of Stable Diffusion
|
| 48 |
width = 512 # default width of Stable Diffusion
|
|
|
|
| 80 |
# And now they're ready!
|
| 81 |
return output
|
| 82 |
|
|
|
|
|
|
|
|
|
|
| 83 |
|
| 84 |
def get_style_embeddings(style_file):
|
| 85 |
style_embed = torch.load(style_file)
|