Hugging Face's logo

Spaces:
x2778
/
flux-dev-multi-lora
Runtime error

App Files Community
jiuface commited on
Commit
df9f472
·
verified ·
1 Parent(s): 969ecbb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,7 +30,7 @@ import diffusers
30
 
31
  # init
32
  dtype = torch.bfloat16
33
- device = "cuda"
34
 
35
  print(device)
36
  base_model = "black-forest-labs/FLUX.1-dev"
@@ -40,7 +40,7 @@ base_model = "black-forest-labs/FLUX.1-dev"
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
 
 
30
 
31
  # init
32
  dtype = torch.bfloat16
33
+ device = "cuda:0"
34
 
35
  print(device)
36
  base_model = "black-forest-labs/FLUX.1-dev"
 
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