added login / ezreal

README.md

@@ -0,0 +1,3 @@
+## Run
+ 1. `pip install transformers==4.19.2 diffusers invisible-watermark`
+ 2. `pip install diffusers["torch"]`

login.py

@@ -0,0 +1,2 @@
+from huggingface_hub import login
+login()

main.py

@@ -3,7 +3,7 @@ from nvidia import execute_nvidia
 
 
 prompt = "ezreal"
-filename = "ezreal22"
+filename = "ezreal221"
 
 
 execute_nvidia(prompt,filename)

nvidia.py

@@ -1,16 +1,15 @@
 from diffusers import StableDiffusionPipeline
+import torch
 
 def execute_nvidia(prompt, filename):
-
     DEVICE = 'cuda'
-
+    model_id = "runwayml/stable-diffusion-v1-5"
     pipe = StableDiffusionPipeline.from_pretrained(
-        "CompVis/stable-diffusion-v1-4",
-        use_auth_token=True
+        model_id,
+        use_auth_token=True,
+        # torch_dtype=torch.float16
     ).to(DEVICE)
-    # prompt = "Elon Musk riding elephant on the moon."
     pipe.enable_attention_slicing()
-    # _ = pipe(prompt, num_inference_steps=1)
     image = pipe(prompt).images[0]
     image.save("pictures/"+ filename +".png")
 

requirements.py

@@ -0,0 +1,3 @@
+# Specify the required packages and their versions
+click==8.1
+transformers==4.19