Spaces:

raman07
/

MIMIC-SD-Demo-Memory-Optimized

Sleeping

Raman Dutt commited on Jan 30, 2024

Commit

d34c7e2

•

1 Parent(s): a2d5835

minor fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -233,7 +233,7 @@ def predict(
     cuda_device = f"cuda:{device}" if torch.cuda.is_available() else "cpu"
     print("Loading Pipeline for {} Fine-Tuning".format(unet_pretraining_type))
-    sd_pipeline_norm = loadSDModel(
         unet_pretraining_type=unet_pretraining_type,
         cuda_device=cuda_device,
     )

     cuda_device = f"cuda:{device}" if torch.cuda.is_available() else "cpu"
     print("Loading Pipeline for {} Fine-Tuning".format(unet_pretraining_type))
+    sd_pipeline = loadSDModel(
         unet_pretraining_type=unet_pretraining_type,
         cuda_device=cuda_device,
     )