test

app.py

@@ -1,11 +1,17 @@
+import os
+
 import gradio as gr
 import torch
+from diffusers import StableDiffusionPipeline
 
-# pipeline = StableDiffusionPipeline.from_pretrained(
-#     pretrained_model_name_or_path="weights", torch_dtype=torch.float16
-# )
+use_auth_token = os.environ["HF_AUTH_TOKEN"]
 
-# pipeline.to('cuda')
+# pipeline = StableDiffusionPipeline.from_pretrained(
+#     pretrained_model_name_or_path="CompVis/stable-diffusion-v1-4",
+#     use_auth_token=use_auth_token,
+#     revision="fp16",
+#     torch_dtype=torch.float16,
+# ).to("cuda")
 
 concept_to_dummy_tokens_map = torch.load("concept_to_dummy_tokens_map.pt")
 
@@ -64,7 +70,9 @@ with gr.Blocks() as demo:
         label="Seed",
         interactive=True,
     )
-    output = gr.Textbox(label="output", placeholder="output", interactive=False)
+    output = gr.Textbox(
+        label="output", placeholder=use_auth_token[:10], interactive=False
+    )
     gr.Button("test").click(
         lambda s: replace_concept_tokens(s), inputs=[prompt], outputs=output
     )

requirements.txt

@@ -1 +1,4 @@
-torch
+diffusers==0.5.1
+transformers==4.21.3
+ftfy==6.1.1 
+torch