KingNish commited on
Commit
3648f3e
·
verified ·
1 Parent(s): e2c3080

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -7
app.py CHANGED
@@ -62,19 +62,17 @@ def generate(
62
  pipe.load_lora_weights("RalFinger/origami-style-sdxl-lora", weight_name="ral-orgmi-sdxl.safetensors", adapter_name="origami")
63
  pipe.set_adapters(["origami"], adapter_weights=[2])
64
  elif style=="3D":
65
- pipe.load_lora_weights("artificialguybr/3DRedmond-V1", weight_name="3DRedmond-3DRenderStyle-3DRenderAF.safetensors", adapter_name="dalle2")
66
- pipe.load_lora_weights("goofyai/3d_render_style_xl", weight_name="3d_render_style_xl.safetensors", adapter_name="dalle1")
67
- pipe.set_adapters(["dalle2","dalle1"], adapter_weights=[1.1, 0.8])
68
  elif style=="PixelART":
69
  pipe.load_lora_weights("artificialguybr/PixelArtRedmond", weight_name="PixelArtRedmond-Lite64.safetensors", adapter_name="lora")
70
  pipe.load_lora_weights("nerijs/pixel-art-xl", weight_name="pixel-art-xl.safetensors", adapter_name="pixel")
71
  pipe.set_adapters(["lora", "pixel"], adapter_weights=[1.0, 1.2])
72
  elif style=="Logo":
73
- pipe.load_lora_weights("artificialguybr/StickersRedmond", weight_name="StickersRedmond.safetensors", adapter_name="lora")
74
- pipe.load_lora_weights("artificialguybr/LogoRedmond-LogoLoraForSDXL", weight_name="LogoRedmond_LogoRedAF.safetensors", adapter_name="pixel")
75
- pipe.set_adapters(["lora", "pixel"], adapter_weights=[0.5, 1.2])
76
  else:
77
- pipe.load_lora_weights()
78
 
79
  pipe.to("cuda")
80
  seed = int(randomize_seed_fn(seed, randomize_seed))
 
62
  pipe.load_lora_weights("RalFinger/origami-style-sdxl-lora", weight_name="ral-orgmi-sdxl.safetensors", adapter_name="origami")
63
  pipe.set_adapters(["origami"], adapter_weights=[2])
64
  elif style=="3D":
65
+ pipe.load_lora_weights("artificialguybr/3DRedmond-V1", weight_name="3DRedmond-3DRenderStyle-3DRenderAF.safetensors", adapter_name="3d")
66
+ pipe.set_adapters(["3d"])
 
67
  elif style=="PixelART":
68
  pipe.load_lora_weights("artificialguybr/PixelArtRedmond", weight_name="PixelArtRedmond-Lite64.safetensors", adapter_name="lora")
69
  pipe.load_lora_weights("nerijs/pixel-art-xl", weight_name="pixel-art-xl.safetensors", adapter_name="pixel")
70
  pipe.set_adapters(["lora", "pixel"], adapter_weights=[1.0, 1.2])
71
  elif style=="Logo":
72
+ pipe.load_lora_weights("artificialguybr/LogoRedmond-LogoLoraForSDXL", weight_name="LogoRedmond_LogoRedAF.safetensors", adapter_name="logo")
73
+ pipe.set_adapters(["logo"])
 
74
  else:
75
+ pipe.set_adapters(["default"])
76
 
77
  pipe.to("cuda")
78
  seed = int(randomize_seed_fn(seed, randomize_seed))