Spaces:

Ehab001
/

texttoimage

Sleeping

Ehab001 commited on Dec 22, 2024

Commit

21758ed

verified ·

1 Parent(s): 4abc347

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,6 @@ import torch
 # Load the model
 model_id = "s3nh/artwork-arcane-stable-diffusion"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
 if pipe.scheduler is not None:
     pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 else:

 # Load the model
 model_id = "s3nh/artwork-arcane-stable-diffusion"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 if pipe.scheduler is not None:
     pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 else: