Hugging Face's logo

Spaces:
wifix199
/
Text_to_image_LuminaIQ
Running

App Files Community
wifix199 commited on
Commit
fe2046f
•
1 Parent(s): 7611e51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
  from diffusers import DiffusionPipeline
4
 
5
  # Load the model with authentication
6
- pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, use_auth_token=True)
7
  pipe.enable_model_cpu_offload() # Save some VRAM by offloading the model to CPU
8
 
9
  def generate_image(prompt):
 
3
  from diffusers import DiffusionPipeline
4
 
5
  # Load the model with authentication
6
+ pipe = DiffusionPipeline.from_pretrained("Shakker-Labs/AWPortrait-FL", torch_dtype=torch.bfloat16, use_auth_token=True)
7
  pipe.enable_model_cpu_offload() # Save some VRAM by offloading the model to CPU
8
 
9
  def generate_image(prompt):