Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files
app.py
CHANGED
@@ -59,7 +59,6 @@ IMAGE_PROCESS_TRANSFORM = transforms.Compose([
|
|
59 |
|
60 |
@spaces.GPU
|
61 |
def generate_image(ref_image, prompt, height=512, width=512, num_steps=25, guidance_scale=3.5, ip_scale=1.0):
|
62 |
-
nonlocal pipe
|
63 |
with torch.no_grad():
|
64 |
image_refs = map(torch.stack, [
|
65 |
[IMAGE_PROCESS_TRANSFORM(i) for i in [ref_image, ]]
|
|
|
59 |
|
60 |
@spaces.GPU
|
61 |
def generate_image(ref_image, prompt, height=512, width=512, num_steps=25, guidance_scale=3.5, ip_scale=1.0):
|
|
|
62 |
with torch.no_grad():
|
63 |
image_refs = map(torch.stack, [
|
64 |
[IMAGE_PROCESS_TRANSFORM(i) for i in [ref_image, ]]
|