elismasilva commited on
Commit
14638d9
·
1 Parent(s): dc528d7

fix load model

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -27,8 +27,7 @@ def load_model(model_id):
27
 
28
  if model_id != last_loaded_model:
29
  pipe = None
30
- torch_gc()
31
-
32
  pipe = StableDiffusionXLControlNetTileSRPipeline.from_pretrained(
33
  model_id, controlnet=controlnet, vae=vae, torch_dtype=torch.float16, use_safetensors=True, variant="fp16"
34
  ).to(device)
 
27
 
28
  if model_id != last_loaded_model:
29
  pipe = None
30
+
 
31
  pipe = StableDiffusionXLControlNetTileSRPipeline.from_pretrained(
32
  model_id, controlnet=controlnet, vae=vae, torch_dtype=torch.float16, use_safetensors=True, variant="fp16"
33
  ).to(device)