Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Update app.py
Browse files
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 =
|
|
|
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 |
|