Hugging Face's logo

Spaces:
AlekseyCalvin
/
soonfactory
Running on Zero

App Files Community
1
AlekseyCalvin commited on
Commit
2fc900f
•
1 Parent(s): 03550a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfold
53
  #).to(device)
54
 
55
  dtype = torch.bfloat16
56
- base_model = "AlekseyCalvin/Artsy_Lite_Flux_v1_by_jurdn_Diffusers"
57
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype).to("cuda")
58
  #pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
59
  torch.cuda.empty_cache()
 
53
  #).to(device)
54
 
55
  dtype = torch.bfloat16
56
+ base_model = "AlekseyCalvin/Shuttle3Diffusion_Diffusers"
57
  pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype).to("cuda")
58
  #pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
59
  torch.cuda.empty_cache()