flux-schnell-lora

Running on Zero

jiuface commited on Oct 11

Commit

df9f472

•

1 Parent(s): 969ecbb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ import diffusers
 # init
 dtype = torch.bfloat16
-device = "cuda"
 print(device)
 base_model = "black-forest-labs/FLUX.1-dev"
@@ -40,7 +40,7 @@ base_model = "black-forest-labs/FLUX.1-dev"
 txt2img_pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype)
 txt2img_pipe = txt2img_pipe.to(device)
-txt2img_pipe.__class__.load_lora_into_transformer = classmethod(load_lora_into_transformer)
 MAX_SEED = 2**32 - 1

 # init
 dtype = torch.bfloat16
+device = "cuda:0"
 print(device)
 base_model = "black-forest-labs/FLUX.1-dev"
 txt2img_pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype)
 txt2img_pipe = txt2img_pipe.to(device)
+# txt2img_pipe.__class__.load_lora_into_transformer = classmethod(load_lora_into_transformer)
 MAX_SEED = 2**32 - 1