Spaces:

ironjr
/

SemanticPalette

Running on Zero

ironjr commited on Mar 19

Commit

ceffafa

•

1 Parent(s): 9ac12f4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -102,7 +102,7 @@ model_dict = {
 }
 models = {
-    k: StableMultiDiffusionPipeline(device, dtype=torch.float32, sd_version='1.5', hf_key=v).cuda()
     for k, v in model_dict.items()
 }

 }
 models = {
+    k: StableMultiDiffusionPipeline(device, sd_version='1.5', hf_key=v).cuda()
     for k, v in model_dict.items()
 }