Krebzonide commited on
Commit
7588cca
·
1 Parent(s): 6346496

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -61,6 +61,7 @@ def set_base_model(base_model_id):
61
  del pipe
62
  torch.cuda.empty_cache()
63
  gc.collect()
 
64
  pipe = StableDiffusionXLPipeline.from_single_file(
65
  model_url,
66
  torch_dtype = torch.float16,
 
61
  del pipe
62
  torch.cuda.empty_cache()
63
  gc.collect()
64
+ model_url = "https://huggingface.co/" + model_url_list[new_model]
65
  pipe = StableDiffusionXLPipeline.from_single_file(
66
  model_url,
67
  torch_dtype = torch.float16,