fallenshock commited on
Commit
400fefc
1 Parent(s): 57c12de
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -93,10 +93,10 @@ def FlowEditRun(
93
 
94
  if model_type == 'FLUX':
95
  # pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.float16)
96
- pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16)
97
  loaded_model = 'FLUX'
98
  elif model_type == 'SD3':
99
- pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
100
  loaded_model = 'SD3'
101
  else:
102
  raise NotImplementedError(f"Model type {model_type} not implemented")
 
93
 
94
  if model_type == 'FLUX':
95
  # pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.float16)
96
+ pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16, token=oauth_token.token)
97
  loaded_model = 'FLUX'
98
  elif model_type == 'SD3':
99
+ pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16, token=oauth_token.token)
100
  loaded_model = 'SD3'
101
  else:
102
  raise NotImplementedError(f"Model type {model_type} not implemented")