Spaces:
Sleeping
Sleeping
change space for hw5
Browse files
app.py
CHANGED
@@ -9,8 +9,8 @@ from diffusers import DiffusionPipeline
|
|
9 |
|
10 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
11 |
# model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like to use
|
12 |
-
model_repo_id = "CompVis/stable-diffusion-v1-4"
|
13 |
-
model_dropdown = ["stabilityai/sdxl-turbo", "CompVis/stable-diffusion-v1-4"]
|
14 |
|
15 |
models = [
|
16 |
"gstranger/kawaiicat-lora-1.4",
|
@@ -39,7 +39,7 @@ def infer(
|
|
39 |
randomize_seed,
|
40 |
width,
|
41 |
height,
|
42 |
-
model_repo_id=model_repo_id,
|
43 |
seed=42,
|
44 |
guidance_scale=7,
|
45 |
num_inference_steps=20,
|
@@ -194,7 +194,7 @@ with gr.Blocks(css=css) as demo:
|
|
194 |
randomize_seed,
|
195 |
width,
|
196 |
height,
|
197 |
-
model_repo_id,
|
198 |
seed,
|
199 |
guidance_scale,
|
200 |
num_inference_steps,
|
|
|
9 |
|
10 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
11 |
# model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like to use
|
12 |
+
# model_repo_id = "CompVis/stable-diffusion-v1-4"
|
13 |
+
# model_dropdown = ["stabilityai/sdxl-turbo", "CompVis/stable-diffusion-v1-4"]
|
14 |
|
15 |
models = [
|
16 |
"gstranger/kawaiicat-lora-1.4",
|
|
|
39 |
randomize_seed,
|
40 |
width,
|
41 |
height,
|
42 |
+
# model_repo_id=model_repo_id,
|
43 |
seed=42,
|
44 |
guidance_scale=7,
|
45 |
num_inference_steps=20,
|
|
|
194 |
randomize_seed,
|
195 |
width,
|
196 |
height,
|
197 |
+
# model_repo_id,
|
198 |
seed,
|
199 |
guidance_scale,
|
200 |
num_inference_steps,
|