Spaces:

hohonu-vicml
/

Trailblazer

Runtime error

eggarsway commited on Feb 5, 2024

Commit

02f0f02

1 Parent(s): 4acbeb5

bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-#import spaces
 import sys
 import time
 import os
@@ -40,7 +40,7 @@ pipe = DiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.to('cuda')
-#@spaces.GPU(duration=120)
 def core(bundle):
     generator = torch.Generator().manual_seed(int(bundle["seed"]))
     result = pipe(

+import spaces
 import sys
 import time
 import os
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.to('cuda')
+@spaces.GPU(duration=120)
 def core(bundle):
     generator = torch.Generator().manual_seed(int(bundle["seed"]))
     result = pipe(