Spaces:

asahi417
/

stable-depth2image-v2

Running on Zero

asahi417 commited on Jul 30

Commit

2be66b4

•

1 Parent(s): b3cdf68

Upload folder using huggingface_hub

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ for n in range(1, 10):
 @spaces.GPU
 def infer(init_image, prompt, negative_prompt, seed, width, height, guidance_scale, num_inference_steps):
-    depth = model_depth.image2depth([init_image])
     return model_image.text2image(
         [init_image],
         depth_maps=depth,

 @spaces.GPU
 def infer(init_image, prompt, negative_prompt, seed, width, height, guidance_scale, num_inference_steps):
+    depth = model_depth.image2depth([init_image], return_tensor=True)
     return model_image.text2image(
         [init_image],
         depth_maps=depth,