Flux-Character-Consistancy

Running on Zero

zs38 commited on Feb 11

Commit

ebd2fe3

1 Parent(s): 03e6b18

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import os
 import torch
-# import spaces
 import safetensors
 import gradio as gr
 from PIL import Image
@@ -56,7 +56,7 @@ IMAGE_PROCESS_TRANSFORM = transforms.Compose([
     transforms.Normalize(mean=[0.4815, 0.4578, 0.4082], std=[0.2686, 0.2613, 0.276])
 ])
-# @spaces.GPU
 def generate_image(ref_image, ref_image2, prompt, height=512, width=512, num_steps=25, guidance_scale=3.5, seed=0, ip_scale=1.0):
     print(f"ref_image: {ref_image.size}, prompt: {prompt}, height: {height}, width: {width}, num_steps: {num_steps}, guidance_scale: {guidance_scale}, ip_scale: {ip_scale}")
     with torch.no_grad():

 import os
 import torch
+import spaces
 import safetensors
 import gradio as gr
 from PIL import Image
     transforms.Normalize(mean=[0.4815, 0.4578, 0.4082], std=[0.2686, 0.2613, 0.276])
 ])
+@spaces.GPU
 def generate_image(ref_image, ref_image2, prompt, height=512, width=512, num_steps=25, guidance_scale=3.5, seed=0, ip_scale=1.0):
     print(f"ref_image: {ref_image.size}, prompt: {prompt}, height: {height}, width: {width}, num_steps: {num_steps}, guidance_scale: {guidance_scale}, ip_scale: {ip_scale}")
     with torch.no_grad():