Spaces:

Tramac
/

style-portrait-v1-5

Runtime error

Tramac commited on Apr 2

Commit

2bc874f

•

1 Parent(s): 61f257d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,8 @@ repo = "Tramac/style-portrait-v1-5"
 pipeline = StableDiffusionPipeline.from_pretrained(
     repo,
     torch_dtype=torch.float16,
-).to("cuda")
 @spaces.GPU(enable_queue=True)
 def generate_image(prompt, neg_prompt):

 pipeline = StableDiffusionPipeline.from_pretrained(
     repo,
     torch_dtype=torch.float16,
+)
+pipeline.enable_model_cpu_offload()
 @spaces.GPU(enable_queue=True)
 def generate_image(prompt, neg_prompt):