Initial commit

.gitattributes

@@ -29,3 +29,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+model/* filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,4 +1,5 @@
 import os
+import pickle
 
 import streamlit as st
 
@@ -12,26 +13,28 @@ st.set_page_config(layout="wide")
 st.title('Play with Stable-Diffusion v1-4')
 
 model_id = "CompVis/stable-diffusion-v1-4"
-# device = "cuda"
+device = "cuda"
 auth_token = os.environ.get("StableDiffusion")
 
 
 with st.spinner(
 	text='Loading...'
 ):
-	pipe = StableDiffusionPipeline.from_pretrained(
-		model_id,
-		revision="fp16",
-		torch_dtype=torch.float16,
-		use_auth_token=auth_token
-	)
-
-	# pipe = pipe.to(device)
+	# pipe = StableDiffusionPipeline.from_pretrained(
+	# 	model_id,
+	# 	revision="fp16",
+	# 	torch_dtype=torch.float16,
+	# 	use_auth_token=auth_token
+	# )
+	with open('model/stable-diffusion.bin', 'rb') as model_file:
+		pipe = pickle.load(model_file)
+		st.success('Model loaded. Start caching to GPU...')
+	pipe = pipe.to(device)
 
 
 def infer(prompt, samples=2, steps=30, scale=7.5, seed=25):
-	# generator = torch.Generator(device=device).manual_seed(seed)
-	generator = torch.Generator().manual_seed(seed)
+	generator = torch.Generator(device=device).manual_seed(seed)
+	# generator = torch.Generator().manual_seed(seed)
 
 	with autocast("cuda"):
 		images_list = pipe(
@@ -62,7 +65,7 @@ with st.form(key='new'):
 
 		steps = col2.select_slider(
 			label='Steps',
-			options=range(1, 61),
+			options=range(1, 101),
 			value=40
 		)
 

model/stable-diffusion.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e6a135cedd2bafde17e9da2138845a83768f62cc0e01a16d29cf5388bc47a3b1
+size 2743182527