Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
import numpy as np
|
3 |
import random
|
4 |
import spaces #[uncomment to use ZeroGPU]
|
5 |
-
from diffusers import DiffusionPipeline ,
|
6 |
import torch
|
7 |
from huggingface_hub import login
|
8 |
import os
|
@@ -18,11 +18,11 @@ else:
|
|
18 |
torch_dtype = torch.float32
|
19 |
|
20 |
pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
|
21 |
-
|
22 |
pipe.load_lora_weights("artificiallover0/man_asshole")
|
23 |
pipe.fuse_lora()
|
24 |
pipe = pipe.to(device)
|
25 |
-
|
26 |
MAX_SEED = np.iinfo(np.int32).max
|
27 |
MAX_IMAGE_SIZE = 1024
|
28 |
|
|
|
2 |
import numpy as np
|
3 |
import random
|
4 |
import spaces #[uncomment to use ZeroGPU]
|
5 |
+
from diffusers import DiffusionPipeline ,
|
6 |
import torch
|
7 |
from huggingface_hub import login
|
8 |
import os
|
|
|
18 |
torch_dtype = torch.float32
|
19 |
|
20 |
pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
|
21 |
+
|
22 |
pipe.load_lora_weights("artificiallover0/man_asshole")
|
23 |
pipe.fuse_lora()
|
24 |
pipe = pipe.to(device)
|
25 |
+
print(pipe.scheduler.compatibles)
|
26 |
MAX_SEED = np.iinfo(np.int32).max
|
27 |
MAX_IMAGE_SIZE = 1024
|
28 |
|