AlekseyCalvin commited on
Commit
a60d56c
·
verified ·
1 Parent(s): f779426

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -47,7 +47,8 @@ with open('loras.json', 'r') as f:
47
  # Initialize the base model
48
  dtype = torch.bfloat16
49
  base_model = "ariG23498/sd-3.5-merged"
50
- pipe = AutoPipelineForText2Image.from_pretrained(base_model, torch_dtype=dtype).to("cuda")
 
51
  #pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
52
  torch.cuda.empty_cache()
53
 
 
47
  # Initialize the base model
48
  dtype = torch.bfloat16
49
  base_model = "ariG23498/sd-3.5-merged"
50
+ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype).to("cuda")
51
+
52
  #pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
53
  torch.cuda.empty_cache()
54