Spaces:

surokpro2
/

sae_flux

Running on Zero

surokpro2 commited on Jun 11

Commit

de759d7

verified ·

1 Parent(s): 2e63856

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -707,6 +707,8 @@ if __name__ == "__main__":
     from SDLens import HookedStableDiffusionXLPipeline
     from SAE import SparseAutoencoder
     from huggingface_hub import hf_hub_download
     dtype = torch.float16
     pipe = EditedFluxPipeline.from_pretrained(

     from SDLens import HookedStableDiffusionXLPipeline
     from SAE import SparseAutoencoder
     from huggingface_hub import hf_hub_download
+    from huggingface_hub import login
+    login(token=os.environ["HF_TOKEN"])
     dtype = torch.float16
     pipe = EditedFluxPipeline.from_pretrained(