Spaces:
Running
on
Zero
Running
on
Zero
fallenshock
commited on
Commit
•
400fefc
1
Parent(s):
57c12de
fix oauth
Browse files
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")
|