gstranger commited on
Commit
66c7a88
·
1 Parent(s): dba807a

change space for hw5

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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,