Spaces:

RiverZ
/

ICEdit

Running on Zero

RiverZ commited on Apr 30

Commit

6f47456

verified ·

1 Parent(s): 795fb65

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ import numpy as np
 import torch
 import argparse
 import random
-from diffusers import FluxFillPipeline
 from PIL import Image
 MAX_SEED = np.iinfo(np.int32).max
@@ -41,7 +41,7 @@ if args.enable_model_cpu_offload:
 else:
     pipe = pipe.to("cuda")
-# @spaces.GPU
 def infer(edit_images,
           prompt,
           seed=666,

 import torch
 import argparse
 import random
+import spaces
 from PIL import Image
 MAX_SEED = np.iinfo(np.int32).max
 else:
     pipe = pipe.to("cuda")
+@spaces.GPU
 def infer(edit_images,
           prompt,
           seed=666,