Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
import spaces
|
|
|
2 |
from peft import PeftModel
|
3 |
import gradio as gr
|
4 |
from diffusers import StableDiffusionImg2ImgPipeline
|
@@ -10,8 +11,10 @@ from diffusers import StableDiffusionPipeline
|
|
10 |
# Load the model
|
11 |
# model_id = "nitrosocke/Ghibli-Diffusion"
|
12 |
# pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
|
|
|
|
|
13 |
model_id = "black-forest-labs/FLUX.1-dev"
|
14 |
-
pipe =DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16,token=
|
15 |
|
16 |
|
17 |
# # 1. 选择一个基础模型,例如 SD 1.5
|
|
|
1 |
import spaces
|
2 |
+
import os
|
3 |
from peft import PeftModel
|
4 |
import gradio as gr
|
5 |
from diffusers import StableDiffusionImg2ImgPipeline
|
|
|
11 |
# Load the model
|
12 |
# model_id = "nitrosocke/Ghibli-Diffusion"
|
13 |
# pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
|
14 |
+
tk=os.getenv('ghtoken')
|
15 |
+
print("ttttt",tk)
|
16 |
model_id = "black-forest-labs/FLUX.1-dev"
|
17 |
+
pipe =DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16,token=tk)
|
18 |
|
19 |
|
20 |
# # 1. 选择一个基础模型,例如 SD 1.5
|