Hugging Face's logo

Spaces:
jiuface
/
flux-schnell-lora
Running on Zero

App Files Community
jiuface commited on
Commit
12b22a2
•
1 Parent(s): 55e274c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,9 +38,9 @@ base_model = "black-forest-labs/FLUX.1-dev"
38
  # load pipe
39
 
40
  txt2img_pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype)
41
- txt2img_pipe.__class__.load_lora_into_transformer = classmethod(load_lora_into_transformer)
42
- txt2img_pipe = txt2img_pipe.to(device)
43
 
 
 
44
 
45
  MAX_SEED = 2**32 - 1
46
 
 
38
  # load pipe
39
 
40
  txt2img_pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype)
 
 
41
 
42
+ txt2img_pipe = txt2img_pipe.to(device)
43
+ txt2img_pipe.__class__.load_lora_into_transformer = classmethod(load_lora_into_transformer)
44
 
45
  MAX_SEED = 2**32 - 1
46