Spaces:

nkanungo
/

Generative_Art_Stable_Diffusion

Runtime error

nkanungo commited on Oct 25, 2023

Commit

2c362e5

1 Parent(s): 1f0a1e3

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -22,7 +22,7 @@ pretrained_model_name_or_path = "segmind/tiny-sd"
 pipe = DiffusionPipeline.from_pretrained(
     pretrained_model_name_or_path,
     low_cpu_mem_usage = True,
-    torch_dtype=torch.float16
 ).to(torch_device)

 pipe = DiffusionPipeline.from_pretrained(
     pretrained_model_name_or_path,
     low_cpu_mem_usage = True,
+    torch_dtype=torch.float32
 ).to(torch_device)