Spaces:
Runtime error
Runtime error
ameerazam08
commited on
Commit
β’
7d47252
1
Parent(s):
684ca13
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
|
|
29 |
|
30 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
31 |
if torch.cuda.is_available():
|
32 |
-
pipe = AutoPipelineForText2Image.from_pretrained('Lykon/dreamshaper-xl-1-0', torch_dtype=torch.float16
|
33 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, use_karras_sigmas=True, algorithm_type="sde-dpmsolver++")
|
34 |
pipe = pipe.to("cuda")
|
35 |
|
|
|
29 |
|
30 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
31 |
if torch.cuda.is_available():
|
32 |
+
pipe = AutoPipelineForText2Image.from_pretrained('Lykon/dreamshaper-xl-1-0', torch_dtype=torch.float16)
|
33 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, use_karras_sigmas=True, algorithm_type="sde-dpmsolver++")
|
34 |
pipe = pipe.to("cuda")
|
35 |
|