Spaces:

surokpro2
/

Unboxing_SDXL_with_SAEs

Running on Zero

surokpro2 commited on 11 days ago

Commit

583b85d

•

1 Parent(s): b8deff7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -367,7 +367,7 @@ if __name__ == "__main__":
     pipe = HookedStableDiffusionXLPipeline.from_pretrained(
         'stabilityai/sdxl-turbo',
         torch_dtype=dtype,
-        device_map="balanced",
         variant=("fp16" if dtype==torch.float16 else None)
     )
     pipe.set_progress_bar_config(disable=True)

     pipe = HookedStableDiffusionXLPipeline.from_pretrained(
         'stabilityai/sdxl-turbo',
         torch_dtype=dtype,
+        device_map="cpu",
         variant=("fp16" if dtype==torch.float16 else None)
     )
     pipe.set_progress_bar_config(disable=True)