Spaces:

dragynir
/

fashion_controlnet

Runtime error

dragynir commited on Feb 2

Commit

755f543

•

1 Parent(s): 85c3ab7

add gpu select

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,8 @@ from src.pipeline import FashionPipeline, PipelineOutput
 config = PipelineConfig()
-fashion_pipeline = FashionPipeline(config, device=torch.device('cuda'))
 def process(

 config = PipelineConfig()
+device = torch.device('cuda') if torch.cuda.is_available() else 'cpu'
+fashion_pipeline = FashionPipeline(config, device=device)
 def process(