Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.b
|
|
38 |
good_vae = AutoencoderKL.from_pretrained("ostris/Flex.1-alpha", subfolder="vae", torch_dtype=torch.bfloat16).to(device)
|
39 |
|
40 |
dtype = torch.bfloat16
|
41 |
-
base_model = "
|
42 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype, use_safetensors=True).to("cuda")
|
43 |
#pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
|
44 |
torch.cuda.empty_cache()
|
|
|
38 |
good_vae = AutoencoderKL.from_pretrained("ostris/Flex.1-alpha", subfolder="vae", torch_dtype=torch.bfloat16).to(device)
|
39 |
|
40 |
dtype = torch.bfloat16
|
41 |
+
base_model = "AlekseyCalvin/FlexAlpha_Soonr"
|
42 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype, use_safetensors=True).to("cuda")
|
43 |
#pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
|
44 |
torch.cuda.empty_cache()
|