Krebzonide
commited on
Commit
·
7588cca
1
Parent(s):
6346496
Update app.py
Browse files
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,
|