Spaces:
Build error
Build error
add origin lora
Browse files- app.py +3 -1
- models/Ghibli.safetensors +3 -0
app.py
CHANGED
|
@@ -36,6 +36,8 @@ def clear_cache(transformer):
|
|
| 36 |
def single_condition_generate_image(prompt, spatial_img, height, width, seed, control_type, use_zero_init, zero_steps):
|
| 37 |
# Set the control type
|
| 38 |
if control_type == "Ghibli":
|
|
|
|
|
|
|
| 39 |
lora_path = os.path.join(lora_base_path, "MaoMu_Ghibli.safetensors")
|
| 40 |
elif control_type == "3D":
|
| 41 |
lora_path = os.path.join(lora_base_path, "3d_animation.safetensors")
|
|
@@ -61,7 +63,7 @@ def single_condition_generate_image(prompt, spatial_img, height, width, seed, co
|
|
| 61 |
return image
|
| 62 |
|
| 63 |
# Define the Gradio interface components
|
| 64 |
-
control_types = ["Ghibli", "3D"]
|
| 65 |
|
| 66 |
# Create the Gradio Blocks interface
|
| 67 |
with gr.Blocks() as demo:
|
|
|
|
| 36 |
def single_condition_generate_image(prompt, spatial_img, height, width, seed, control_type, use_zero_init, zero_steps):
|
| 37 |
# Set the control type
|
| 38 |
if control_type == "Ghibli":
|
| 39 |
+
lora_path = os.path.join(lora_base_path, "Ghibli.safetensors")
|
| 40 |
+
elif control_type == "MaoMu_Ghibli":
|
| 41 |
lora_path = os.path.join(lora_base_path, "MaoMu_Ghibli.safetensors")
|
| 42 |
elif control_type == "3D":
|
| 43 |
lora_path = os.path.join(lora_base_path, "3d_animation.safetensors")
|
|
|
|
| 63 |
return image
|
| 64 |
|
| 65 |
# Define the Gradio interface components
|
| 66 |
+
control_types = ["Ghibli", "MaoMu_Ghibli", "3D"]
|
| 67 |
|
| 68 |
# Create the Gradio Blocks interface
|
| 69 |
with gr.Blocks() as demo:
|
models/Ghibli.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a5204551bd9cb587d659fe1cc50cf524b6339348bc5b1c3ea3b4efe71eb5e753
|
| 3 |
+
size 298895992
|