Spaces:

Alibrown
/

FluxUC-MotionSynt

Runtime error

Alibrown commited on 20 days ago

Commit

4e69a41

verified ·

1 Parent(s): 2e49e0c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 pipe = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
-pipe.load_lora_weights('enhanceaiteam/Flux-uncensored', weight_name='lora.safetensors')
 pipe = pipe.to(device)
 MAX_SEED = np.iinfo(np.int32).max

 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 pipe = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe.load_lora_weights('enhanceaiteam/Flux-uncensored')
 pipe = pipe.to(device)
 MAX_SEED = np.iinfo(np.int32).max