Update app.py
Browse files
app.py
CHANGED
|
@@ -96,7 +96,7 @@ STYLE_NAMES = list(styles.keys())
|
|
| 96 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 97 |
|
| 98 |
#sampling_schedule = AysSchedules["StableDiffusionXLTimesteps"]
|
| 99 |
-
upscaler = UpscaleWithModel.from_pretrained("
|
| 100 |
|
| 101 |
def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
|
| 102 |
if style_name in styles:
|
|
|
|
| 96 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 97 |
|
| 98 |
#sampling_schedule = AysSchedules["StableDiffusionXLTimesteps"]
|
| 99 |
+
upscaler = UpscaleWithModel.from_pretrained("ford442/ClearRealityV1").to(torch.device("cuda:0"))
|
| 100 |
|
| 101 |
def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
|
| 102 |
if style_name in styles:
|