Skip to content

GitLab

Commit aae34b1f authored by chenpangpang's avatar chenpangpang
Browse files

feat: 去除spaces

parent 9a73404f
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
PhotoMaker/gradio_demo/app.py
View file @ aae34b1f
...@@ -8,7 +8,6 @@ from diffusers.utils import load_image ...@@ -8,7 +8,6 @@ from diffusers.utils import load_image
from diffusers import EulerDiscreteScheduler from diffusers import EulerDiscreteScheduler
from huggingface_hub import hf_hub_download from huggingface_hub import hf_hub_download
import spaces
import gradio as gr import gradio as gr
from photomaker import PhotoMakerStableDiffusionXLPipeline from photomaker import PhotoMakerStableDiffusionXLPipeline
...@@ -60,8 +59,6 @@ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config) ...@@ -60,8 +59,6 @@ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
pipe.fuse_lora() pipe.fuse_lora()
pipe.to(device) pipe.to(device)
@spaces.GPU
def generate_image(upload_images, prompt, negative_prompt, aspect_ratio_name, style_name, num_steps, def generate_image(upload_images, prompt, negative_prompt, aspect_ratio_name, style_name, num_steps,
style_strength_ratio, num_outputs, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)): style_strength_ratio, num_outputs, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
# check the trigger word # check the trigger word
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment