Hugging Face's logo

Spaces:
NikhilJoson
/
Add-it
Runtime error

App Files Community
NikhilJoson commited on
Commit
b5329fc
·
verified ·
1 Parent(s): 2bbf193

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ MAX_SEED = np.iinfo(np.int32).max
44
  DEVICE = "cuda" #if torch.cuda.is_available() else "cpu"
45
 
46
  inpaint_pipe = FluxInpaintPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
47
-
48
 
49
 
50
  @spaces.GPU()
 
44
  DEVICE = "cuda" #if torch.cuda.is_available() else "cpu"
45
 
46
  inpaint_pipe = FluxInpaintPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
47
+ inpaint_pipe.load_lora_weights('hugovntr/flux-schnell-realism', weight_name='schnell-realism_v1')
48
 
49
 
50
  @spaces.GPU()