Spaces:

AideepImage
/

360_Virtual_Image_prediction

Runtime error

AideepImage commited on Aug 3

Commit

5149010

•

1 Parent(s): 84b71f7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ model_path = snapshot_download("archerfmy0831/sd-t2i-360panoimage")
 # Initialize pipelines
 txt2panoimg = Text2360PanoramaImagePipeline(model_path, torch_dtype=torch.float16)
-@spaces.GPU(duration=200)
 def text_to_pano(prompt, upscale):
     input_data = {'prompt': prompt, 'upscale': upscale, 'refinement': False}
     output = txt2panoimg(input_data)

 # Initialize pipelines
 txt2panoimg = Text2360PanoramaImagePipeline(model_path, torch_dtype=torch.float16)
+@spaces.CPU(duration=200)
 def text_to_pano(prompt, upscale):
     input_data = {'prompt': prompt, 'upscale': upscale, 'refinement': False}
     output = txt2panoimg(input_data)