Abdullah-Habib commited on
Commit
05fd8e7
·
1 Parent(s): 6815e34

testing complete

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ with open('loras.json', 'r') as f:
23
  loras = json.load(f)
24
 
25
  # Initialize the base model
26
- base_model = "stabilityai/stable-diffusion-xl-refiner-1.0"
27
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.float16)
28
  pipe.to("cuda")
29
 
 
23
  loras = json.load(f)
24
 
25
  # Initialize the base model
26
+ base_model = "stabilityai/stable-diffusion-xl-base-1.0"
27
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.float16)
28
  pipe.to("cuda")
29