Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,14 +24,14 @@ from diffusers import AutoPipelineForText2Image
|
|
| 24 |
from diffusers.pipelines.wuerstchen.pipeline_wuerstchen_prior import DEFAULT_STAGE_C_TIMESTEPS
|
| 25 |
|
| 26 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 27 |
-
|
| 28 |
assert os.path.exists("wuerstchen")
|
| 29 |
pipe = AutoPipelineForText2Image.from_pretrained(Path("wuerstchen"), local_files_only = True,
|
| 30 |
torch_dtype=torch.float32)
|
| 31 |
'''
|
| 32 |
pipe = AutoPipelineForText2Image.from_pretrained("warp-ai/wuerstchen",
|
| 33 |
torch_dtype=torch.float32)
|
| 34 |
-
|
| 35 |
pipe.to(device)
|
| 36 |
|
| 37 |
pipe.safety_checker = None
|
|
|
|
| 24 |
from diffusers.pipelines.wuerstchen.pipeline_wuerstchen_prior import DEFAULT_STAGE_C_TIMESTEPS
|
| 25 |
|
| 26 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 27 |
+
'''
|
| 28 |
assert os.path.exists("wuerstchen")
|
| 29 |
pipe = AutoPipelineForText2Image.from_pretrained(Path("wuerstchen"), local_files_only = True,
|
| 30 |
torch_dtype=torch.float32)
|
| 31 |
'''
|
| 32 |
pipe = AutoPipelineForText2Image.from_pretrained("warp-ai/wuerstchen",
|
| 33 |
torch_dtype=torch.float32)
|
| 34 |
+
|
| 35 |
pipe.to(device)
|
| 36 |
|
| 37 |
pipe.safety_checker = None
|