moyanwang
commited on
Commit
•
bb99b60
1
Parent(s):
fb79cc3
delete unuse code
Browse files- lyraSD/muse_trt/models.py +1 -11
lyraSD/muse_trt/models.py
CHANGED
@@ -53,17 +53,7 @@ class BaseModel():
|
|
53 |
assert latent_height >= self.min_latent_shape and latent_height <= self.max_latent_shape
|
54 |
assert latent_width >= self.min_latent_shape and latent_width <= self.max_latent_shape
|
55 |
return (latent_height, latent_width)
|
56 |
-
|
57 |
-
min_batch = batch_size if static_batch else self.min_batch
|
58 |
-
max_batch = batch_size if static_batch else self.max_batch
|
59 |
-
latent_height = image_height // 8
|
60 |
-
latent_width = image_width // 8
|
61 |
-
min_latent_height = latent_height if static_shape else self.min_latent_shape
|
62 |
-
max_latent_height = latent_height if static_shape else self.max_latent_shape
|
63 |
-
min_latent_width = latent_width if static_shape else self.min_latent_shape
|
64 |
-
max_latent_width = latent_width if static_shape else self.max_latent_shape
|
65 |
-
return (min_batch, max_batch, min_latent_height, max_latent_height, min_latent_width, max_latent_width)
|
66 |
-
|
67 |
|
68 |
class CLIP(BaseModel):
|
69 |
def get_model(self):
|
|
|
53 |
assert latent_height >= self.min_latent_shape and latent_height <= self.max_latent_shape
|
54 |
assert latent_width >= self.min_latent_shape and latent_width <= self.max_latent_shape
|
55 |
return (latent_height, latent_width)
|
56 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
57 |
|
58 |
class CLIP(BaseModel):
|
59 |
def get_model(self):
|